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) -- 輸入陣列。

回傳值:

輸出陣列。

回傳值型別:

array