cuquantum.cutensornet.contraction_optimize¶
- cuquantum.cutensornet.contraction_optimize(intptr_t handle, intptr_t desc_net, intptr_t optimizer_config, uint64_t workspace_size_constraint, intptr_t optimizer_info)[源代码]¶
为给定的张量网络计算“优化”的收缩顺序以及切片信息(更多信息请参阅概述部分),以便在遵守用户提供的内存约束的同时,最大限度地缩短总求解时间。
- 参数
handle (intptr_t) – 持有 cuTensorNet 库上下文的不透明句柄。
desc_net (intptr_t) – 描述张量网络的拓扑结构(即,所有张量、它们的连通性和模式)。
optimizer_config (intptr_t) – 保存所有超优化参数,这些参数控制着“最优”收缩顺序的搜索。
workspace_size_constraint (uint64_t) – 用户将提供的最大设备内存(即,cuTensorNet 必须在此用户定义的约束内找到可行的路径/切片解决方案)。
optimizer_info (intptr_t) – 返回时,此对象将保存有关优化路径和相关切片信息的所有必要信息。
optimizer_info
将包含的信息包括(请参阅cutensornetContractionOptimizerInfoAttributes_t
):。