| Safe Haskell | Safe-Inferred |
|---|---|
| Language | Haskell2010 |
Torch.Distributions.Bernoulli
Documentation
Instances
| Show Bernoulli Source # | |
| Distribution Bernoulli Source # | |
Defined in Torch.Distributions.Bernoulli Methods batchShape :: Bernoulli -> [Int] Source # eventShape :: Bernoulli -> [Int] Source # expand :: Bernoulli -> [Int] -> Bernoulli Source # support :: Bernoulli -> Constraint Source # mean :: Bernoulli -> Tensor Source # variance :: Bernoulli -> Tensor Source # sample :: Bernoulli -> [Int] -> IO Tensor Source # logProb :: Bernoulli -> Tensor -> Tensor Source # | |
fromLogits :: Tensor -> Bernoulli Source #