我有一个在pytorch 0.3.1上运行良好的代码,但在最新版本上却有bug。 当我运行代码时,它告诉我这个错误:
RuntimeError: shape mismatch: value tensor of shape [16] cannot be broadcast to indexing result of shape [16, 1]
在旧版本中,我没有问题,在比较时,我使用的是这种格式(版本0.3.1 )
> self.values[oldest_indices] : torch.Size([16, 1])
> self.values[oldest_indices] :
> 4
> 7
> 13
> 16
> 20
> 27
> 31
> 38
> 40
> 45
> 54
> 59
> 62
> 69
> 74
> 78
> [torch.cuda.LongTensor of size 16x1 (GPU 0)]
但是在新版本的Pytorch中,我遇到了以上错误。 当我重塑时,我有这个,但我不想重塑,因为会出现其他错误。 (最新版本):
> self.values[oldest_indices] : torch.Size([16, 1])
> self.values[oldest_indices] : tensor([[ 4],
> [ 6],
> [12],
> [18],
> [22],
> [27],
> [34],
> [37],
> [40],
> [49],
> [51],
> [57],
> [60],
> [66],
> [71],
> [77]], device='cuda:0')
如何获得与旧版本相同的格式? 预先感谢您的帮助:)