728x90
728x90
Tensor들이 저장된 list를 Tensor로 바꾸는 법이다.
torch.tensor(tensor list)로 바꿔도 되지만, 이렇게 하면 비효율적이라나 뭐라나 경고문이 뜬다.
이때 아래와 같은 방법으로 바꾸게 되면 아주 잘 수행된다!
tensor_list = torch.stack(tensor_list, dim=0)728x90
728x90
'연구 > PyTorch' 카테고리의 다른 글
| [PyTorch] tensor 확장하기: torch.expand vs torch.repeat vs torch.repeat_interleave (0) | 2024.02.21 |
|---|---|
| [Pytorch] Dataloader와 Sampler (0) | 2023.10.07 |
| [PyTorch] Tensor 조작법 기본) indexing, view, squeeze, unsqueeze (0) | 2023.09.17 |
| PyTorch로 AutoEncoder 구현하기 (0) | 2023.08.16 |
| PyTorch DataLoader 사용하기 & epoch, batch, iteration 개념 (0) | 2023.08.03 |