import torch

x = torch.FloatTensor([[1,2], [3,4]])
y = torch.FloatTensor([[5,6], [7,8]])
print('========= x =========')
print(x)
print('========= y =========')
print(y)

print('========= concat by dim 0 =========')
print(torch.cat([x, y], dim=0))
print('========= concat by dim 1 =========')
print(torch.cat([x, y], dim=1))

• dim 0 으로 concat을 하면 행을 기준으로 concat
  • 2x2, 2x2 tensor들을 4x2
• dim 1 으로 concat을 하면 열을 기준으로 concat
  • 2x2, 2x2 tensor들을 2x4