cunumeric.linalg.multi_dot#

cunumeric.linalg.multi_dot(arrays, *, out=None)#

Compute the dot product of two or more arrays in a single function call, while automatically selecting the fastest evaluation order. multi_dot chains dot and uses optimal parenthesization of the matrices.

Parameters

arrays (Sequence[array_like]) – If the first argument is 1-D it is treated as a row vector. If the last argument is 1-D it is treated as a column vector. The other arguments must be 2-D.
out (ndarray, optional) – Output argument. This must have the same shape and dtype that would be returned if it was not used.

Returns

output – Returns the dot product of the supplied arrays.

Return type

ndarray