mlx.core.einsum

einsum(subscripts: str, *operands, stream: None | Stream | Device = None) → array

Perform the Einstein summation convention on the operands.

參數:

• subscripts (str) -- The Einstein summation convention equation.

• *operands (array) -- The input arrays.

回傳:

The output array.

回傳型別:

array