sample_prob_array#

cuquantum.bindings.custabilizer.sample_prob_array( intptr_t handle, int64_t num_samples, int64_t num_probs, intptr_t probs, uint64_t seed, intptr_t samples, intptr_t stream, )[source]#

Sample Bernoulli random bits from a probability array.

Parameters:

handle (intptr_t) – Library handle.
num_samples (int64_t) – Number of samples (minor dimension). Must be a multiple of 32.
num_probs (int64_t) – Number of probabilities (major dimension).
probs (intptr_t) – Probability array of length num_probs (device-accessible pointer). Values should be in [0, 1]; out-of-range values are clamped, NaN is treated as 0.
seed (uint64_t) – Random seed.
samples (intptr_t) – Output buffer for bit-packed samples (device-accessible pointer). Must have at least (num_probs * num_samples / 32) words.
stream (intptr_t) – CUDA stream.