如何将二维张量重塑为一维

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]

如何将二维张量重塑为一维

How can reshape 2d tensor to 1d

python

reshape

conv-neural-network

pytorch