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