http://sjh4773.github.io/categories/pytorch-%EA%B3%B5%EB%B6%80/ Recent content in Pytorch 공부 on My New Hugo Site Hugo -- gohugo.io kr-ko Sun, 28 Mar 2021 12:36:58 +0000 http://sjh4773.github.io/post/python-2/ Sun, 28 Mar 2021 12:36:58 +0000 http://sjh4773.github.io/post/python-2/ cat 함수란? cat함수는 concatenate를 해주는 함수이고 concatenate하고자 하는 차원을 증가시킨다. concatenate하고자하는 차원을 지정해주면 그 차원으로 두 tensor의 차원을 더한 값으로 차원이 변경된다. cat 함수의 시각화 Python 코드 import torch batch_size, N, K = 3, 10, 256 x = torch.rand(batch_size, N, K) # [M, N, K] y = torch.rand(batch_size, N, K) # [M, N, K] output1 = torch.cat([x,y], dim=1) #[M, N+N, K] output2 = torch.cat([x,y], dim=2) #[M, N, K+K]