cupy.matmul

cupy.matmul = <cupy._core._gufuncs._GUFunc object>

matmul(x1, x2, /, out=None, **kwargs)

两个数组的矩阵乘积。

返回两个数组的矩阵乘积，它是 Python 3.5 中根据 PEP465 引入的 @ 运算符的实现。

与 cupy.dot 的主要区别在于对维度大于 2 的数组的处理。更多信息请参见 numpy.matmul()。

参数:

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

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

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

• **kwargs – ufunc 关键字参数。

返回:

输出数组。

返回类型:

cupy.ndarray

另请参阅

numpy.matmul()