250x250
Notice
Recent Posts
Recent Comments
Link
목록unsqueeze (1)
코딩걸음마
[딥러닝] Pytorch 차원다루기(Shaping/squeeze)
1. Tensor shape 파악하기 (.shape / .size() ) Tensor를 눈으로 보고 바로 몇차원이며 몇개의 원소가 들어가 있는지 한번에 파악하기에는 현실적으로 너무 어렵다. 하지만, 한번에 파악할 수 있는 방법이 다행히(?) 있다. x = torch.FloatTensor([[[1, 2], [3, 4]], [[5, 6], [7, 8]], [[9, 10], [11, 12]]]) print(x.shape) print(x.size()) torch.Size([3, 2, 2]) torch.Size([3, 2, 2]) .shape .size() 를 입력하면 torch.Size([ 3 ,2 , 2])라고 나오는데 이를 도식화 하면 다음과 같다 1-2. Tensor shape 변경하기 (reshape) ..
딥러닝_Pytorch
2022. 6. 23. 20:33