반응형 분류 전체보기733 [pytorch] torch.stack() 텐서합치는 방법 텐서를 합치는 방법으로 torch.stack 함수를 사용하는 방법이 있습니다. torch.stack함수를 사용하면 축이 하나 늡니다. 자세한 원리는 아래에서 보겠습니다. 목차torch.stack 함수에 대하여?stack 함수 사용 시 지정하는 축으로 차원을 확장하여 tensor를 쌓는 것을 의미합니다. tensor를 쌓기 때문에 축이 하나 늘죠. 텐서를 차곡차곡 쌓는 것이기 때문에 두 tensor의 차원이 정확하게 일치하여야만 torch.stack 함수를 사용할 수 있습니다. 그리고 stack을 할 때 새로 추가할 축을 dim 입력변수로 정해주어야 한다. torch.stack 시각화 torch.stack 코드import torchB, H, W = 10, 20, 128x1 = torch.rand(.. 2024. 9. 29. 이전 1 ··· 211 212 213 214 215 216 217 ··· 733 다음 반응형
