cuquantum.cutensornet.marginal_compute

cuquantum.cutensornet.marginal_compute(intptr_t handle, intptr_t tensor_network_marginal, projected_mode_values, intptr_t work_desc, intptr_t marginal_tensor, intptr_t cuda_stream)[源代码]

计算请求的张量网络状态边缘张量。

参数
  • handle (intptr_t) – cuTensorNet 库句柄。

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

  • projected_mode_values (object) –

    指向投影模式值的指针。每个整数值对应于给定(投影)状态模式的基态。它可以是

    • 一个指向数组的指针地址的 int,或者

    • 一个 int64_t 的 Python 序列。

  • work_desc (intptr_t) – 工作空间描述符(所需 scratch/cache 内存缓冲区必须由用户设置)。

  • marginal_tensor (intptr_t) – 指向边缘张量的 GPU 存储的指针,该张量将在此调用中计算。

  • cuda_stream (intptr_t) – CUDA 流。