View

3주차 PyTorch Basics

2021.08.17

PyTorch Basics Tensor 다차원 Arrays를 표현하는 Pytorch 클래스 사실상 numpy의 ndarray와 동일(Tensorflow의 Tensor와도 동일) Tensor를 생성하는 함수도 거의 동일 numpy는 ndarray라는 객체를 사용하고 pytorch에서는 tensor라는 객체를 사용 example #numpy - ndarray import numpy as np n_array = np.arange(10).reshape(2,5) print(n_array) print("n_dim:",n_array.ndim, "shape: ",n_array.shape) -> [[0 1 2 3 4] [5 6 7 8 9]] n_dim: 2 shape: (2, 5) ---------------------..

부스트캠프 AI Tech 2기/Precourse

Tensor Manipulation 2

2021.07.29

Pytorch View numpy의 reshape와 동일 t = np.array([[[0,1,2],[3,4,5]], [[6,7,8],[9,10,11]]] ft = torch.FloatTensor(t) ft -> tensor([[[ 0., 1., 2.], [ 3., 4., 5.]], [[ 6., 7., 8.], [ 9., 10., 11.]]]) ft.shape -> torch.Size([2, 2, 3]) ft.view([-1,3]).shape -> torch.Size([4, 3]) ft.view([4,1,3]).shape -> torch.Size([4, 1, 3]) Squeeze 차원이 1인 경우에는 해당 차원을 제거함 ft = torch.FloatTensor([[0],[1],[2]]) ft.shape -..

3주차 PyTorch Basics

Tensor Manipulation 2

티스토리툴바