- 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.
If it is an integer, then
axesaxes at the last of
aand the first of
If it is a pair of sequences of integers, then these two sequences specify the list of axes for
b. The corresponding axes are paired for sum-product.
The tensor dot product of
balong the axes specified by
- Return type