|TensorFlow 1 version||View source on GitHub|
Tensor contraction over specified indices and outer product.
See Migration guide for more details.
tf.einsum( equation, *inputs, **kwargs )
Used in the notebooks
|Used in the guide||Used in the tutorials|
Einsum allows defining Tensors by defining their element-wise computation.
This computation is defined by
equation, a shorthand form based on Einstein
summation. As an example, consider multiplying two matrices A and B to form a
matrix C. The elements of C are given by:
C[i,k] = sum_j A[i,j] * B[j,k]
The corresponding einsum
In general, to convert the element-wise equation into the