description: TensorFlow variant of NumPy's tensordot.

tf.experimental.numpy.tensordot

View source on GitHub

TensorFlow variant of NumPy's tensordot.

tf.experimental.numpy.tensordot(
    a, b, axes=2
)

See the NumPy documentation for numpy.tensordot.