jax-js
    Preparing search index...

    Function tensordot

    • Compute the tensor dot product of two N-dimensional arrays.

      The behavior is determined by axes. If an integer k, sum over the last k axes of x and the first k axes of y. If a tuple, then the first array corresponds to the axes of x and the second to the axes of y.

      Parameters

      Returns Array