public class Bernoulli extends java.lang.Object implements Distribution<BooleanTensor>
Modifier and Type | Method and Description |
---|---|
DoubleTensor |
dLogProb(BooleanTensor x) |
DoubleTensor |
logProb(BooleanTensor x) |
static DoubleVertex |
logProbGraph(LogProbGraph.BooleanPlaceholderVertex x,
LogProbGraph.DoublePlaceholderVertex probTrue) |
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 static DoubleVertex logProbGraph(LogProbGraph.BooleanPlaceholderVertex x, LogProbGraph.DoublePlaceholderVertex probTrue)
public DoubleTensor dLogProb(BooleanTensor x)