public class Bernoulli extends java.lang.Object implements Distribution<BooleanTensor>
Modifier and Type | Method and Description |
---|---|
DoubleTensor |
dLogProb(BooleanTensor x) |
DoubleTensor |
logProb(BooleanTensor x) |
BooleanTensor |
sample(long[] shape,
KeanuRandom random) |
static Bernoulli |
withParameters(DoubleTensor probTrue) |
public static Bernoulli withParameters(DoubleTensor probTrue)
public BooleanTensor sample(long[] shape, KeanuRandom random)
sample
in interface Distribution<BooleanTensor>
public DoubleTensor logProb(BooleanTensor x)
logProb
in interface Distribution<BooleanTensor>
public DoubleTensor dLogProb(BooleanTensor x)