ValueError: Expected parameter scale of distribution Normal to satisfy the constraint GreaterThan
ValueError: Expected parameter scale (Tensor of shape (2, 2)) of distribution Normal(loc: torch.Size([2, 2]), scale: torch.Size([2, 2])) to satisfy the constraint GreaterThan(lower_bound=0.0), but fou
·
所遇到的错误:
相应的代码:
import torch
import torch.distributions as dist
p = dist.Normal
mu = torch.Tensor(torch.randn(2, 2))
logvar = torch.Tensor(torch.randn(2, 2))
pz_x = p(mu, logvar) # 在这一步报错
z = pz_x.rsample(torch.Size([1]))
测试发现不是type的问题,发现将logvar的生成语句改为下面的方式就不会报错:
logvar = torch.Tensor(torch.rand(2, 2))
- torch.randn() 用来生成满足标准正态分布的随机数(有正有负)
- torch.rand() 用来生成满足0-1的均匀分布的一组随机数
查看了torch.distributions.Normal的官方文档,并对比上面error所提示的 GreaterThan(lower_bound=0.0),可以知道logvar这里的参数需要输入的是正数,所以用rand不会报错,而randn会:
开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!
更多推荐
已为社区贡献5条内容
所有评论(0)