通过连接从元组的多个元素创建火炬张量的简单方法

如何解决通过连接从元组的多个元素创建火炬张量的简单方法

输入

我有以下列表

r1 = [([[[1,2,3],[1,3]],[[4,5,6],[4,6]]],[[[7,8],[7,8]],[[9,10],[9,10]]]),([[[11,12,13],[11,13]],[[14,15,16],[14,16]]],[[[17,18],[17,18]],[[19,20],[19,20]]])]

我将根据上面的输入生成 2 个火炬张量。

我想要的输出如下

输出

output = 
[tensor([[[ 1,[ 1,[[ 4,[ 4,6]],[[11,16]]]),tensor([[[ 7,[ 7,[[ 9,[ 9,10]],[[17,20]]])]

我的代码如下。

output = []
for i in range(len(r1[0])):
    templates = []
    for j in range(len(r1)):
        templates.append(torch.tensor(r1[j][i]))
        template = torch.cat(templates)
    output.append(template)

是否有更简单或更容易的方法来获得我想要的结果？

解决方法

这样做：

output = [torch.Tensor([*a,*b]) for a,b in zip(*r1)]

它先连接两个列表的对应项，然后创建张量