Data is ___ ?
article thumbnail
[Pytorch] 텐서(차원) 관련 함수 정리
Python/라이브러리 2023. 1. 18. 21:06

✨ 계속 추가 예정 🔎 텐서 합치기 torch.cat concatenate 해주는 함수로써 concatenate하고자 하는 차원을 증가시킨다. (단, 차원의 수는 유지) concatenate하고자 하는 차원을 지정해주면 그 차원으로 두 텐서의 차원을 더한 값으로 변경 dimension 지정해주지 않으면 default=0으로 설정 (예를 들어) 10x2, 10x2인 두 텐서가 있을 때, cat 기준이 0이면 20x2, 1이면 10x4 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([..

profile on loading

Loading...