在這裏先創建一個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], dim=0) # 结果是 tensor([
Xue Li