x = x.view(-1, 320)代表参数表示的含义

在torch代码中经常遇到x.view,一般第一个参数都设置成-1。
代表可以是任意值,不是固定的。
举个简单的例子:
a = torch.arange(1, 17)

a.view(4, 4) # output below
tensor([[1, 2, 3, 4],
[5, 6, 7, 8],
[9, 10, 11, 12],
[13, 14, 15, 16]])
[torch.FloatTensor of size 4x4]
运行结果为:
在这里插入图片描述
等价于:

a.view(-1,4)
结果与上面的相同。
发现一个问题:x=torch.arange(1,17)说明x的 shape is (16,),所以输入的view的第二个参数必须被16整除,因此当前面第一个参数设置-1的情况下,结果也是确定的。

猜你喜欢

转载自blog.csdn.net/weixin_43979572/article/details/84958633
x