cuquantum.cutensornet.state_prepare

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

准备完整张量网络状态的计算，可以采用精确形式或分解形式。

参数

  • handle (intptr_t) – cuTensorNet 库句柄。

  • tensor_network_state (intptr_t) – 张量网络状态。

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

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

  • cuda_stream (intptr_t) – CUDA 流。

参见

cutensornetStatePrepare()