torch.flatten(input, start_dim=0, end_dim=-1) → Tensor
>>> t = torch.tensor([[[1, 2],
[3, 4]],
[[5, 6],
[7, 8]]])
>>> torch.flatten(t)
tensor([1, 2, 3, 4, 5, 6, 7, 8])
>>> torch.flatten(t, start_dim=1)
tensor([[1, 2, 3, 4],
[5, 6, 7, 8]])
(계속 정리...)
반응형
'데이터 과학 > 딥러닝 FrameWork' 카테고리의 다른 글
RNN & LSTM 설명 및 구현(pytorch) (0) | 2021.03.29 |
---|---|
모델 앙상블(ensemble) 하기 (0) | 2021.02.26 |
Trouble Shooting (0) | 2021.01.31 |
Torch 데이터셋 & 데이터 로더 + Transforms (0) | 2021.01.30 |
Pytorch with examples (0) | 2020.12.14 |