cuquantum.cutensornet.marginal_prepare

cuquantum.cutensornet.marginal_prepare(intptr_t handle, intptr_t tensor_network_marginal, size_t max_workspace_size_device, intptr_t work_desc, intptr_t cuda_stream)[源代码]

准备计算请求的张量网络状态边际张量。

参数

  • handle (intptr_t) – cuTensorNet 库句柄。

  • tensor_network_marginal (intptr_t) – 张量网络状态边际表示。

  • max_workspace_size_device (size_t) – 可用 GPU 暂存内存量的上限（字节）。

  • work_desc (intptr_t) – 工作区描述符（将设置所需的暂存/缓存内存大小）。

  • cuda_stream (intptr_t) – CUDA 流。

另请参阅

cutensornetMarginalPrepare()