PyTorch 기본 함수 정리 (Type Casting)
2022. 6. 18. 12:18ㆍPyTorch
lt = torch.LongTensor([1, 2, 3, 4])
print(lt)
# tensor([1, 2, 3, 4])
print(lt.float())
# tensor([1., 2., 3., 4.])
# ByteTensor: Boolean을 저장
bt = torch.ByteTensor([True,False,False,True])
print(bt)
print(bt.long())
print(bt.float())
# tensor([1, 0, 0, 1], dtype=torch.uint8)
# tensor([1, 0, 0, 1])
# tensor([1., 0., 0., 1.])
lt == 3
# tensor([False, False, True, False])'PyTorch' 카테고리의 다른 글
| PyTorch 기본 함수 정리(In-place Operation) (0) | 2022.06.18 |
|---|---|
| PyTorch 기본 함수 정리 (Ones, Zeros) (0) | 2022.06.18 |
| PyTorch 기본 함수 정리 (Concatenate, Stack) (0) | 2022.06.18 |
| PyTorch 기본 함수 정리 (View, Squeeze, Unsqueeze) (0) | 2022.06.18 |
| PyTorch 기본 함수 정리 (mean, max) (0) | 2022.06.18 |