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

10 lines
175 B
Python

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