description: TensorFlow variant of NumPy's einsum.

tf.experimental.numpy.einsum

View source on GitHub

TensorFlow variant of NumPy's einsum.

tf.experimental.numpy.einsum(
    subscripts, *operands, **kwargs
)

See the NumPy documentation for numpy.einsum.