cuquantum.cutensornet.create_contraction_plan

cuquantum.cutensornet.create_contraction_plan(intptr_t handle, intptr_t desc_net, intptr_t optimizer_info, intptr_t work_desc) → intptr_t[source]

初始化 cutensornetContractionPlan_t。

参数

  • handle (intptr_t) – 持有 cuTensorNet 库上下文的不透明句柄。

  • desc_net (intptr_t) – 描述张量网络（即，其张量及其连通性）。

  • optimizer_info (intptr_t) – 不透明结构。

  • work_desc (intptr_t) – 描述工作区的不透明结构。在创建收缩计划时，仅需要工作区大小；指向工作区内存的指针可以留空。如果设置了设备内存处理程序，则 work_desc 可以设置为 null（在这种情况下，将推断“推荐”工作区大小，请参阅 CUTENSORNET_WORKSIZE_PREF_RECOMMENDED）或设置为有效的 cutensornetWorkspaceDescriptor_t，其中设置了所需的工作区大小，并且工作区指针为空，请参阅 内存管理 API 部分。

返回

cuTensorNet 的收缩计划保存执行张量收缩所需的所有信息；准确地说，它为每个需要收缩整个张量网络的张量收缩初始化一个 cutensorContractionPlan_t。

返回类型

intptr_t

参见

cutensornetCreateContractionPlan()