Pytorch迁移0.3.1到1.0-大家好

时间:2019-02-28 13:50:54

标签: python python-3.x pytorch

我有一个在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')

如何获得与旧版本相同的格式? 预先感谢您的帮助:)

0 个答案:

没有答案