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()