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.

  • 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.


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

Return type: