cuquantum.cutensornet.state_compute

cuquantum.cutensornet.state_compute(intptr_t handle, intptr_t state, intptr_t workspace, state_tensors_out, intptr_t stream) → tuple

计算张量网络状态表示。

参数

• handle (intptr_t) – 库句柄。

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

• workspace (intptr_t) – 工作区描述符。

• state_tensors_out –

  每个输出张量（在设备上）的指针地址（作为 Python int）的主机数组。它可以是

  • 作为数组指针地址的 int

  • Python int 序列

• stream (intptr_t) – CUDA 流句柄（cudaStream_t 作为 Python int）。

返回值

输出张量的元数据：(extents_out, strides_out)。

返回类型

tuple

参见

cutensornetStateCompute()