scvi.distributions.NegativeBinomial.sample_n

NegativeBinomial.sample_n(n)

Generates n samples or n batches of samples if the distribution parameters are batched.