cupy.dot

cupy.dot(a, b, out=None)

返回两个数组的点积。

对于轴数多于一的数组，它计算沿 a 的最后一个轴和 b 的倒数第二个轴的点积。如果两个数组都是二维的，这只是一个矩阵乘法。对于一维数组，它使用其唯一轴作为计算点积的轴。

参数:

• a (cupy.ndarray) – 左侧参数。

• b (cupy.ndarray) – 右侧参数。

• out (cupy.ndarray) – 输出数组。

返回:

a 和 b 的点积。

返回类型:

cupy.ndarray

另请参阅

numpy.dot()