不是concat的意思
import torch
a = torch.ones([1,2])
b = torch.ones([1,2])
torch.stack([a,b],1)
(0 ,.,.) =
1 1
1 1
[torch.FloatTensor of size 1x2x2]
不是concat的意思
import torch
a = torch.ones([1,2])
b = torch.ones([1,2])
torch.stack([a,b],1)
(0 ,.,.) =
1 1
1 1
[torch.FloatTensor of size 1x2x2]