[머신러닝/딥러닝] Pytorch 기본 함수 cat stack expand repeat randperm argmax topk squeeze dim 차원 행 unsqueeze 축소

지난 시간에는 PyTorch 를 사용하기 위해 필요한 기본 함수들에 대해 알아봤습니다. https://blog.naver.com/dorergiverny/223315492761 [머신러닝/딥러닝] PyTorch 함수 정리 메소드 구글 코랩 Google Colab 개발 환경 확인 CUDA cuDNN OS 브로드캐스팅 split index_select 지난 시간에 우리는 딥러닝에서의 주축을 이루고 있는 대표적인 모델 및 기술 경향에 대해 간략히 알아봤습... blog.naver.com 이번에는 지난 시간에 이어 PyTorch의 기본 함수들을 좀 더 정리해보겠습니다. cat cat 메소드는 tensor를 행 방향으로, 열 방향으로 이어 붙이는 메소드 입니다. 이는 concatenate의 약자입니다. 이는 dim의 수는 유지가 되면서 행 또는 열 방향으로 늘어나게 됩니다. 이는 network 에서 다른 두 feature를 fusion 할 때와 batch 단위로 생성된 output 값을 모아 ...

원문링크 : [머신러닝/딥러닝] Pytorch 기본 함수 cat stack expand repeat randperm argmax topk squeeze dim 차원 행 unsqueeze 축소