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 ofa
and the first ofb
are used.If it is a pair of sequences of integers, then these two sequences specify the list of axes for
a
andb
. The corresponding axes are paired for sum-product.
- Returns:
The tensor dot product of
a
andb
along the axes specified byaxes
.- Return type:
See also