cuquantum.cutensornet.marginal_prepare¶

cuquantum.cutensornet.marginal_prepare(intptr_t handle, intptr_t marginal, size_t max_workspace_size_device, intptr_t workspace, intptr_t stream)[source]¶

Prepares computation of the tensor network state marginal distribution.

Parameters

handle (intptr_t) – The library handle.
marginal (intptr_t) – The tensor network marginal computation handle.
max_workspace_size_device (size_t) – The maximal device workspace size (in bytes) allowed for the mariginal computation.
workspace (intptr_t) – The workspace descriptor.
stream (intptr_t) – The CUDA stream handle (cudaStream_t as Python int).