Searched refs:einsum (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/ops/
H A Dspecial_math_ops_test.py239 _ = 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 Dspecial_math_ops.py87 @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 Dlinalg_impl.py41 einsum = special_math_ops.einsum variable
/external/tensorflow/tensorflow/contrib/kfac/python/ops/
H A Dfisher_factors.py1001 case_wise_gradients = special_math_ops.einsum("bijk,bijl->bkl", patches,

Completed in 234 milliseconds