Tensordot

Returns a contraction of a and b over multiple dimensions.

Abstract Signature:

Tensordot(a: Tensor, b: Tensor, dims: Union[int, Tuple] = 2)

PyTorch

API: torch.tensordot
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.tensordot
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.tensordot
Strategy: Direct Mapping
Official Docs ↗

Keras

API: keras.ops.tensordot
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: tf.tensordot
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.tensordot
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

API: jax.numpy.tensordot
Strategy: Direct Mapping
Official Docs ↗

PaxML / Praxis

API: jax.numpy.tensordot
Strategy: Direct Mapping
Official Docs ↗