def _sample(self, n_samples):
p = tf.sigmoid(self.logits)
shape = tf.concat([[n_samples], self.batch_shape], 0)
alpha = tf.random_uniform(
shape, minval=0, maxval=1, dtype=self.param_dtype)
samples = tf.cast(tf.less(alpha, p), dtype=self.dtype)
static_n_samples = n_samples if isinstance(n_samples, int) else None
samples.set_shape(
tf.TensorShape([static_n_samples]).concatenate(
self.get_batch_shape()))
return samples
评论列表
文章目录