a=torch.Tensor(range(1,7)).view(2,3)
b=torch.Tensor(range(7,13)).view(2,3)
c=torch.Tensor(range(13,19)).view(2,3)
d=torch.stack((a,b,c),dim=0)
print(a)
print(b)
print(c)
print(d)
########## dim=0
tensor([[1., 2., 3.],
[4., 5., 6.]])
tensor([[ 7., 8., 9.],
[10., 11., 12.]])
tensor([[13., 14., 15.],
[16., 17., 18.]])
tensor([[[ 1., 2., 3.],
[ 4., 5., 6.]],
[[ 7., 8., 9.],
[10., 11., 12.]],
[[13., 14., 15.],
[16., 17., 18.]]])
######### dim=1
tensor([[1., 2., 3.],
[4., 5., 6.]])
tensor([[ 7., 8., 9.],
[10., 11., 12.]])
tensor([[13., 14., 15.],
[16., 17., 18.]])
tensor([[[ 1., 2., 3.],
[ 7., 8., 9.],
[13., 14., 15.]],
[[ 4., 5., 6.],
[10., 11., 12.],
[16., 17., 18.]]])
########## dim=2
tensor([[1., 2., 3.],
[4., 5., 6.]])
tensor([[ 7., 8., 9.],
[10., 11., 12.]])
tensor([[13., 14., 15.],
[16., 17., 18.]])
tensor([[[ 1., 7., 13.],
[ 2., 8., 14.],
[ 3., 9., 15.]],
[[ 4., 10., 16.],
[ 5., 11., 17.],
[ 6., 12., 18.]]])