numpy.tensordot(a, b, axes=2) → ndarray
cupy.tensordot(a, b, axes=2) → ndarray
dask.array.tensordot(lhs, rhs, axes=2) → ndarray
jax.numpy.tensordot(a, b, axes=2, *, precision=None) → ndarray
np.tensordot(a, b, axes=2) → ndarray
torch.tensordot(a, b, dims=2) → Tensor
tf.tensordot(a, b, axes, name=None) → Tensor
Allows providing axes as a (2,K) shaped array, in addition to an integer or tuple. Returned data type is that of a.