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:

cupy.ndarray

See also

numpy.matmul()