Searched refs:einsum (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/python/ops/ |
H A D | special_math_ops_test.py | 239 _ = special_math_ops.einsum(axes, *inputs) 247 _ = special_math_ops.einsum( 262 _ = special_math_ops.einsum(axes, *inputs) 275 output_tensor = special_math_ops.einsum(axes, *input_tensors) 280 correct_value = np.einsum(axes, *input_vals) 290 out = special_math_ops.einsum('ij,jk->ik', m0, m1) 302 out = special_math_ops.einsum('ij,j->i', m0, m1) 314 out = special_math_ops.einsum('ijk,kl->ijl', m0, m1) 326 out = special_math_ops.einsum('kl,ijk->ijl', m0, m1) 338 out = special_math_ops.einsum('ij [all...] |
H A D | special_math_ops.py | 87 @tf_export('einsum', 'linalg.einsum') 88 def einsum(equation, *inputs, **kwargs): function 117 >>> einsum('ij,jk->ik', m0, m1) # output[i,k] = sum_j m0[i,j] * m1[j, k] 120 >>> einsum('i,i->', u, v) # output = sum_i u[i]*v[i] 123 >>> einsum('i,j->ij', u, v) # output[i,j] = u[i]*v[j] 126 >>> einsum('ij->ji', m) # output[j,i] = m[i,j] 129 >>> einsum('aij,ajk->aik', s, t) # out[a,i,k] = sum_j s[a,i,j] * t[a, j, k] 132 This function behaves like `numpy.einsum`, but does not support: 141 `numpy.einsum` [all...] |
/external/tensorflow/tensorflow/python/ops/linalg/ |
H A D | linalg_impl.py | 41 einsum = special_math_ops.einsum variable
|
/external/tensorflow/tensorflow/contrib/kfac/python/ops/ |
H A D | fisher_factors.py | 1001 case_wise_gradients = special_math_ops.einsum("bijk,bijl->bkl", patches,
|
Completed in 234 milliseconds