cuquantum.cutensornet.sampler_prepare¶

cuquantum.cutensornet.sampler_prepare(intptr_t handle, intptr_t sampler, 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.
sampler (intptr_t) – The tensor network sampler.
max_workspace_size_device (size_t) – The maximal device workspace size (in bytes) allowed for the sampling computation.
workspace (intptr_t) – The workspace descriptor.
stream (intptr_t) – The CUDA stream handle (cudaStream_t as Python int).