AI/pytorch
-
max, gatherAI/pytorch 2021. 6. 28. 23:12
https://pytorch.org/docs/stable/generated/torch.max.html 의 내용이며, 각 인자의 설명 해석은 틀릴수 있음에 유의 torch.max(input) - input tensor의 모든 요소중 최대값을 반환 사용 예제 a = torch.randn(1,3) torch.max(a) 또는 a.max() torch.max(input, dim, keepdim=False, *, out=None) - 주어진 dim의 차원을 갖는 input이라는 텐서의 각 열(row)에서의 최대값과 해당 인덱스를 튜플로 반환 (values, indices) - keepdim이 True라면, 각 차원의 size가 1인 경우를 제외하고 output 텐서는 input과 동일한 크기를 같는다 - keep..
-
Tensor 변환과 결합 - cat(), view(), squeeze(), unsqueeze()AI/pytorch 2021. 6. 28. 20:57
본 내용은 https://blog.naver.com/qbxlvnf11/221627488337 에서 대부분 발췌해 온 것임 Pytorch에서 텐서를 결합할 때는 cat() 함수를 활용, dim 파라미터를 통해 기준을 설정 - torch 라이브러리 import import torch - 테스트용 텐서 생성 a = torch.ones(4, 3) b = torch.zeros(3, 3) - 텐서 열 기준으로 concatenate (각 텐서의 열의 수가 같아야 함) torch.cat([a, b], dim=0) - 테스트용 텐서 생성 a = torch.ones(4, 3) b = torch.zeros(4, 4) - 텐서 행 기준으로 concatenate (각 텐서의 행의 수가 같아야 함) torch.cat([a, ..
-
squeeze(), unsqueeze()함수와 주의점AI/pytorch 2021. 6. 28. 20:50
squeeze함수 - 차원을 줄일때 사용 squeeze함수는 차원이 1인 차원을 제거해준다. 따로 차원을 설정하지 않으면 1인 차원을 모두 제거한다. 그리고 차원을 설정해주면 그 차원만 제거한다. 주의할 점은 생각치도 못하게 batch가 1일 때 batch차원도 없애버리는 불상사가 발생할 수있다 unsqueeze함수 - 차원을 늘릴때 사용 unsqueeze함수는 squeeze함수의 반대로 1인 차원을 생성하는 함수이다. 그래서 어느 차원에 1인 차원을 생성할 지 꼭 지정해주어야한다.