PyTorch 기본 함수 정리 (View, Squeeze, Unsqueeze)

View (Numpy의 Reshape) t = np.array([[[0, 1, 2], [3, 4, 5]], [[6, 7, 8], [9, 10, 11]]]) t array([[[ 0, 1, 2], [ 3, 4, 5]], [[ 6, 7, 8], [ 9, 10, 11]]]) t.shape (2, 2, 3) 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]) print(ft.view(-1,3).shape) ft.view(-1,3) torch.Size([4, 3]) tensor([[ 0., 1., 2.], [ 3., 4., 5..

2022.06.18