mlx.core.einsum

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

对操作数执行爱因斯坦求和约定。

参数：

• subscripts (str) – 爱因斯坦求和约定公式。

• *operands (array) – 输入数组。

返回值：

输出数组。

返回类型：

array