我有一个torch.tensor
看起来像这样的:
tensor([[[A,B,C],
[D,E,F],
[G,H,I]],
[[J,K,L],
[M,N,O],
[P,Q,R]]]
我想重塑这个张量,使其尺寸为(18, 1)
. 我希望新张量看起来像这样:
tensor([A,
J,
B,
K,
C,
L,
D,
M,
...
I,
R])
我试过tensor.view(-1,1)
了,但这不起作用..