cupy.tensordot

cupy.tensordot(a, b, axes=2)

返回两个数组沿指定轴的张量点积。

这等同于将指定轴重塑为单个轴后计算点积。

参数：

• a (cupy.ndarray) – 第一个参数。

• b (cupy.ndarray) – 第二个参数。

• axes –

  • 如果它是一个整数，则使用 a 的最后 axes 个轴和 b 的最前 axes 个轴。

  • 如果它是一对整数序列，则这两个序列分别指定 a 和 b 的轴列表。相应的轴将进行配对以计算求和乘积。

返回值：

沿由 axes 指定的轴计算 a 和 b 的张量点积。

返回类型：

cupy.ndarray

另请参阅

numpy.tensordot()