view改变维度(只要每个维度相乘不变即可?)
输入
import torch
a = torch.tensor([[[1, 2, 3], [4, 5, 6]],
[[1, 2, 3], [4, 5, 6]]])
print(a.shape)
a1 = a.view(2, 1, 2, 3)
print(a1.shape)
print(a1)
#a1 = a.view(1, 4, 1, 3)可行
#a1 = a.view(4, 3)
输出
torch.Size([2, 2, 3])
torch.Size([2, 1, 2, 3])
tensor([[[[1, 2, 3],
[4, 5, 6]]],
[[[1, 2, 3],
[4, 5, 6]]]])
``view改变维度,只要每个维度相乘不变即可
可以增加维度,减少维度?