js_reverse/ayf_ocr/02pytorch基础语法/数据类型转换一.py
2023-05-26 13:34:28 +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)