mlx.core.einsum_path

目录

mlx.core.einsum_path#

einsum_path(subscripts: str, *operands)#

Compute the contraction order for the given Einstein summation.

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

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

返回:

The einsum path and a string containing information about the chosen path.

返回类型:

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