반응형 분류 전체보기733 [Pytorh,파이토치] torch.cat, 텐서합치는 방법 pytorch에서 텐서를 합치는 방법으로 torch.cat과 torch.stack이 있는데 이번 글에서는 torch.cat에 대해서만 알아보자. 목차 torch.cat 함수에 대하여? orch.cat 함수를 이용하면 텐서들을 concatenate 해준다. concatenate하면서 텐서의 축은 늘어나지않고 이미 있던 축의 차원을 늘려준다. concatenate할 때 축을 지정하면 그 축으로 두 tensor의 축에 해당하는 차원의 합만큼 축의 차원이 변경된다. concatenate 할 때 dimensión 변수를 따로 넣어주지 않으면 default로 0으로 설정된다. 뉴럴넷 내의 다른 두 feature를 fusion하거나 batch 단위의 output을 모아 모든 데이터셋에 대한 output을 저장할 때 .. 2024. 9. 28. 이전 1 ··· 215 216 217 218 219 220 221 ··· 733 다음 반응형