jax.numpy.linalg.tensordot

Contents

jax.numpy.linalg.tensordot#

jax.numpy.linalg.tensordot(x1, x2, /, *, axes=2)[source]#
Parameters:
  • x1 (ArrayLike) –

  • x2 (ArrayLike) –

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

Return type:

Array