cupy.matmul¶
-
cupy.
matmul
(ndarray a, ndarray b, ndarray out=None) → ndarray¶ Returns the matrix product of two arrays and is the implementation of the @ operator introduced in Python 3.5 following PEP465.
The main difference against cupy.dot are the handling of arrays with more than 2 dimensions. For more information see
numpy.matmul()
.Note
The out array as input is currently not supported.
- Parameters
a (cupy.ndarray) – The left argument.
b (cupy.ndarray) – The right argument.
out (cupy.ndarray) – Output array.
- Returns
Output array.
- Return type
See also
numpy.matmul()