在 pytorch 中,如何快速从一个均值和协方差矩阵已知的多元正态分布中采样多个向量?
问题背景
这涉及到正态总体的无偏估计,不记得的同学请出门左拐重修概率论(狗头)
问题分析
样本均值和样本离差阵都可以通过 torch
的内置方法,比如按列求和快速获得,那么问题就划归为,在 pytorch 中,如何快速从一个均值和协方差矩阵已知的多元正态分布中采样多个向量?
torch 的内置多元正态分布模块 distributions
其实 pytorch 本身就提供了生成多元正态分布的模块。假设我们需要采样的正态分布为:
mu = torch.FloatTensor([1, 2, 0])
sigma = torch.FloatTensor([
[2, 0, 0],
[0, 5, 0],
[0, 0,