jax.numpy.linalg.tensordot

jax.numpy.linalg.tensordot(x1, x2, /, *, axes=2)

Parameters:

• x1 (jax.typing.ArrayLike)

• x2 (jax.typing.ArrayLike)

• axes (int | tuple[Sequence[int], Sequence[int]])

Return type:

Array