js_reverse/机器学习/02pytorch基础语法/数据类型转换二.py
2023-07-12 23:26:55 +08:00

13 lines
206 B
Python

import torch
list_2 = [
[127, 128, 3, 4, 5, 6],
[1, 2, 3, 4, 5, 6],
[1, 2, 3, 4, 5, 6],
]
t2 = torch.tensor(list_2) # [-128 - 127]
print(t2.int())
print(t2.double())
print(t2.double().dtype)