当前位置:   article > 正文

Torch.Distributions.Normal

torch.distributions.normal

Pytorch提供的正态分布,官方文档
创建方式

>>> m = Normal(torch.tensor([0.0]), torch.tensor([1.0]))
>>> m.sample()  # 生成数据,可传shape参数
        tensor([ 0.1046])
  • 1
  • 2
  • 3

接口log_prob(self, value)

返回value概率的log值,正态分布公式为
f ( x ) = 1 2 π σ e x p ( − ( x − μ ) 2 2 σ 2 ) f(x)=\frac{1}{\sqrt{2\pi}\sigma}exp(-\frac{(x-\mu)^2}{2\sigma^2}) f(x)=2π σ1exp(2σ2(xμ)2)
函数的计算公式为:
l o g ( f ( x ) ) = − l o g ( 2 π ) − l o g ( σ ) − ( x − μ ) 2 2 σ 2 log(f(x)) = -log(\sqrt{2\pi})-log(\sigma)-\frac{(x-\mu)^2}{2\sigma^2} log(f(x))=log(2π )log(σ)2σ2(xμ)2

接口def entropy(self)

返回值公式
1 2 + 2 π 2 + l o g ( σ ) \frac{1}{2}+\frac{2\pi}{2}+log(\sigma) 21+22π+log(σ)
看着就是正态分布公式的log值,但不知为什么多个 1 2 \frac{1}{2} 21

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/weixin_40725706/article/detail/313979
推荐阅读
相关标签
  

闽ICP备14008679号