mlx.core.einsum_path

einsum_path(subscripts: str, *operands)

计算给定爱因斯坦求和的收缩顺序。

参数:

• subscripts (str) – 爱因斯坦求和约定方程。

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

返回:

einsum 路径和包含所选路径信息的字符串。

返回类型:

tuple(list(tuple(int, int)), str)