[Pytorch] Tensor, unsqueeze, squeeze, view


[Pytorch] Tensor, unsqueeze, squeeze, view

Tensor수학의 벡터나 행렬을 일반화한 개념랭크(Rank, 차원) 0 - 스칼라 / 랭크 1 - 벡터 / 랭크 2 - 2차원 행렬 / 랭크 3 - 3차원 행렬shape, size(), ndimension() 함수로 차원의 크기와 텐서의 랭크를 확인할 수 있다텐서의 랭크(차원) 변경아래 함수들은 원소 수를 그대로 유지하며 차원과 모양만 조절한다.torch.unsqueeze(input, dim)특정 위치에 1인 차원을 추가해 텐서의 차원을 늘리는 함수실행 결과는 다음과 같다torch.squeeze(input, dim=None, *, out=None)dim 을 지정해주지 않으면 텐서의 모든 1인 차원을 제거한다위와 같이 두 가지 방법으로 사용할 수 있다. 아래는 실행 결과이다. unsqueeze로 추가했던 1인 차..........



원문링크 : [Pytorch] Tensor, unsqueeze, squeeze, view