以如下 tensor a
为例,展示常用的维度变换操作
>>> a = torch.rand(4,3,28,28)
>>> a.shape
torch.Size([4, 3, 28, 28])
-
view / reshape 两者功能完全相同:
a.view(shape)
>>> a.view(4,3,28*28) ## a.view(4,3,2
以如下 tensor a
为例,展示常用的维度变换操作
>>> a = torch.rand(4,3,28,28)
>>> a.shape
torch.Size([4, 3, 28, 28])
view / reshape 两者功能完全相同: a.view(shape)
>>> a.view(4,3,28*28) ## a.view(4,3,2