cupy.tensordot#

cupy.tensordot(a, b, axes=2)[source]#

Returns the tensor dot product of two arrays along specified axes.

This is equivalent to compute dot product along the specified axes which are treated as one axis by reshaping.

Parameters:
  • a (cupy.ndarray) – The first argument.

  • b (cupy.ndarray) – The second argument.

  • axes

    • If it is an integer, then axes axes at the last of a and the first of b are used.

    • If it is a pair of sequences of integers, then these two sequences specify the list of axes for a and b. The corresponding axes are paired for sum-product.

Returns:

The tensor dot product of a and b along the axes specified by axes.

Return type:

cupy.ndarray