반응형
list에 tensor가 담겨져 있는 경우, list를 없애고 torch.tensor로 바꾸는 방법은 아래와 같다.
# segm_result = list[tensor]
segm_result = torch.stack(segm_result, dim=0)
반응형
'Python > PyTorch 공부' 카테고리의 다른 글
| [PyTorch] register_hook을 사용하여 Transformer 내부의 Attention matrix(Torch.Tensor)의 gradient 받아오기 (0) | 2022.07.14 |
|---|---|
| [Pytorch] Sementation mask 시각화 하기 (0) | 2022.06.26 |
| [PyTorch] 학습 progress bar 설정하기 (14) | 2022.05.17 |
| COCOstuff dataset class name (1) | 2022.02.09 |
| VOC class name (1) | 2022.02.09 |