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 einsum path and a string containing information about the chosen path.

回傳值型別:

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