Как создать булевский тензор, значения которого взяты из распределения Бернулли?

Question

Я хотел бы реализовать следующую функцию:

def bernoulli_tensor(shape, probability) -> tf.Tensor:
    ...

Answer 1

Вот одна из возможных реализаций:

from typing import List

import tensorflow as tf
import tensorflow_probability as tfp



def bernoulli(shape: List[int], truth_probability: float = 0.5) -> tf.Tensor:
    distribution = tfp.distributions.Bernoulli(probs=truth_probability)
    return tf.cast(distribution.sample(shape), dtype=tf.dtypes.bool)