js_reverse/ayf_ocr/02pytorch基础语法/张量数据合并拉升.py
2023-05-26 13:34:28 +08:00

18 lines
328 B
Python

import torch
list_8 = [[[792, 436, 928, 303],
[809, 170, 778, 652],
[967, 520, 419, 184]],
[[402, 754, 327, 979],
[713, 926, 879, 934],
[540, 953, 209, 369]]]
t8 = torch.tensor(list_8)
# 合并成单列单行
print(t8.view(-1))
# 拉升成4行6列
print(t8.view(4, 6))