cuquantum.einsum_path

cuquantum.einsum_path(subscripts, *operands, optimize=True)

用于评估使用 cuTensorNet 的“最佳”收缩顺序的 numpy.einsum_path() 的直接替代品。

此接口仅支持 NumPy 选项的子集。contract_path() 函数提供了特定于 cuTensorNet 的广泛选项集，建议优先使用此函数。

爱因斯坦求和表达式支持显式和隐式形式。除了下标格式外，交错格式也受支持，作为指定操作数及其模式标签的一种手段。有关操作数类型以及示例的更多详细信息，请参阅 Network。

参数

• subscripts – 模式标签（下标），将爱因斯坦求和表达式定义为逗号分隔的字符序列。表达式中允许使用 Unicode 字符，从而扩展了可以使用爱因斯坦求和约定指定的张量网络的大小。

• operands – 张量（类 ndarray 对象）序列。当前支持的类型为 numpy.ndarray、cupy.ndarray 和 torch.Tensor。

• optimize – 此参数指定路径优化选项。此接口唯一允许的值为 True。

返回值

一个 2 元组 (path, opt_info)

• path一个列表，以字符串 ‘einsum_path’ 开头，后跟一系列操作数序号对

  表示 numpy.einsum_path() 格式的最佳收缩顺序。

• opt_info类型为 OptimizerInfo 的对象的字符串表示形式，包含有关

  最佳收缩顺序的信息。

返回类型

tuple