torch_bernoulli {torch} | R Documentation |
Bernoulli
Description
Bernoulli
Usage
torch_bernoulli(self, p, generator = NULL)
Arguments
self |
(Tensor) the input tensor of probability values for the Bernoulli distribution |
p |
(Number) a probability value. If |
generator |
( |
bernoulli(input, *, generator=NULL, out=NULL) -> Tensor
Draws binary random numbers (0 or 1) from a Bernoulli distribution.
The input
tensor should be a tensor containing probabilities
to be used for drawing the binary random number.
Hence, all values in input
have to be in the range:
.
The element of the output tensor will draw a
value
according to the
probability value given
in
input
.
The returned out
tensor only has values 0 or 1 and is of the same
shape as input
.
out
can have integral dtype
, but input
must have floating
point dtype
.
Examples
if (torch_is_installed()) {
a = torch_empty(c(3, 3))$uniform_(0, 1) # generate a uniform random matrix with range c(0, 1)
a
torch_bernoulli(a)
a = torch_ones(c(3, 3)) # probability of drawing "1" is 1
torch_bernoulli(a)
a = torch_zeros(c(3, 3)) # probability of drawing "1" is 0
torch_bernoulli(a)
}
[Package torch version 0.13.0 Index]