Skip to main content

//clika-runtime/io.clika.runtime/Ops/bernoulli

bernoulli

[common]
fun bernoulli(probabilities: Tensor, device: Placement? = null): Tensor

bernoulli(probabilities: Tensor, device: Placement? = null): the bernoulli operator. Independent Bernoulli draws from per-element success probabilities. Each output element is 1 with probability probabilities[i], else 0; shape and dtype mirror probabilities.