[英]How can reshape 2d tensor to 1d
tensor([[0, 5],
[1, 4],
[2, 2],
[4, 2],
[7, 9],
[2, 0])
我想将这个张量重塑为
tensor([[5],
[14],
[22],
[42],
[79],
[20])
我怎么解决这个问题? 请帮我
我不知道张量数据类型,但对于普通列表,您可以遍历每个内部列表并将它们转换为数字
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])
output 将是:
[5, 14, 22, 42, 79, 20, 0]
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.