在這裏先創建一個4*4二維张量,想象下一個正方形由很多個小方塊組成。在這裏 dim 的值為0時,合并的是行,為1時合并的是列。下面生成了三個相同的二維張量,做拼接。tensor = torch.ones(4, 4)
t1 = torch.cat([tensor, tensor, tensor], dim=1)
print(t1)輸出tensor([[1., 0., 1., 1., 1., 0., 1., 1., 1., 0., 1., 1.],
[1., 0., 1., 1., 1., 0., 1., 1., 1., 0., 1., 1.],
[1., 0., 1., 1., 1., 0., 1., 1., 1., 0., 1., 1.],
[1., 0., 1., 1., 1., 0., 1., 1., 1., 0., 1., 1.]])那麽到了一維,這裏是 0 ,結果并不是[[1,2],[3,4]]a = torch.tensor([1, 2])
b = torch.tensor([3, 4])
torch.cat([a, b], d
Xue Li