什么是nensy.random.multivariate_normal的Tensorflow等价物？

Question

我想绘制给定均值和协方差矩阵的随机样本。在numpy我可以使用numpy.random.multivariate_normal函数来做到这一点。什么是Tensorflow相当于此？

就我而言，数据是2-D。因此，平均值至少是维数的张量（batch_size，1），平均列表的长度为2。

Answer 1

我认为您正在寻找tf.contrib.distributions.MultivariateNormalDiag，如TensorFlow Official Documentation中所示。

Answer 2

要获取随机样本，可以使用tf.distributions.Normal类的方法sample。代替协方差矩阵，它需要标准偏差。例如使用二维：

mean = [1.0, 2.0]
std = [0.5, 0.7]
dist = tf.distributions.Normal(mean, std)
dist.sample()