cuquantum.cutensornet.workspace_compute_contraction_sizes

cuquantum.cutensornet.workspace_compute_contraction_sizes(intptr_t handle, intptr_t desc_net, intptr_t optimizer_info, intptr_t work_desc)[源代码]

计算使用提供的收缩路径收缩输入张量网络所需的工作区大小。

参数

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

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

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

  • work_desc (intptr_t) – 在其中收集信息的工作区描述符。

另请参阅

cutensornetWorkspaceComputeContractionSizes()