如何将2d张量重塑为1d

Question

tensor([[0, 5],
[1, 4],
[2, 2],
[4, 2],
[7, 9],
[2, 0]) 

我想把这个张量重塑为

tensor([[5],
[14],
[22],
[42],
[79],
[20])

我该如何解决这个问题？请帮我

Answer 1

我不知道张量数据类型，但对于普通列表，您可以迭代每个内部列表，并将其转换为数字

def toDigit(l):
z = 1
s = 0
for v in l[::-1]:
s += v * z
z *= 10
return s

a = [[0, 5],[1, 4],[2, 2],[4, 2],[7, 9],[2, 0], [0,0]]
print([toDigit(t) for t in a])

输出为：

[5, 14, 22, 42, 79, 20, 0]