pytorch 报错noise =noise.data.reshape(data.size()[0],input_dim,1,1).normal_(0,1)RuntimeError: shape ‘[32, 100, 1, 1]’ is invalid for input of size 6400
原因： 维度写错了
解决办法： 由于我是dataloader加载的数据，所以加上属性drop_last = True ,
noise = noise.data.reshape(data.size()[0],input_dim,1,1).normal_(0,1) RuntimeError: shape '[32, 100,
猜你喜欢
转载自blog.csdn.net/weixin_43914889/article/details/104677581
今日推荐
周排行