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.
- If it is an integer, then
- out (cupy.ndarray) – Output array.
Returns: The tensor dot product of
a
andb
along the axes specified byaxes
.Return type: See also