所遇到的错误:
在这里插入图片描述
相应的代码:

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会:
在这里插入图片描述

Logo

开放原子开发者工作坊旨在鼓励更多人参与开源活动,与志同道合的开发者们相互交流开发经验、分享开发心得、获取前沿技术趋势。工作坊有多种形式的开发者活动,如meetup、训练营等,主打技术交流,干货满满,真诚地邀请各位开发者共同参与!

更多推荐