js_reverse/ayf_ocr/02pytorch基础语法/获取张量形状.py
2023-05-26 13:34:28 +08:00

12 lines
179 B
Python

import torch
list_3 = [
[127, 128, 3, 4, 5, 6],
[1, 2, 3, 4, 5, 6],
[1, 2, 3, 4, 5, 6],
]
t3 = torch.tensor(list_3) # [-128 - 127]
print(t3.shape)
print(t3.size())