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
Returns

Output array.

Return type

cupy.ndarray

See also

numpy.matmul()