torch.cat()
torch.cat() 함수는 원하는 dimension 방향으로 텐서를 나란하게 쌓아준다.
사용 방법은 다음과 같다.
1
2
output = torch.cat([x1, x2], dim=1)
output = torch.cat([x1, x2], dim=0)
[PyTorch] torch.cat()
This post is licensed under CC BY 4.0 by the author.
torch.cat() 함수는 원하는 dimension 방향으로 텐서를 나란하게 쌓아준다.
사용 방법은 다음과 같다.
1
2
output = torch.cat([x1, x2], dim=1)
output = torch.cat([x1, x2], dim=0)
A new version of content is available.