在Torch中，最惯用的张量组合方式是什么？

2015-7-2 13:58:36
收藏：0
阅读：71
评论：1

我需要将三个张量连接在一起，使3个px1张量成为一个3px1张量。

我能想到最简洁的方法是：

torch.Tensor{v2:totable(),v4:totable(),v6:totable()}:view(3*p,1)

有没有不需要转换为表格再转回张量的方法呢？如果它们的形状兼容，似乎应该有一种通用的方法来连接张量。

我可以想象编写这样的函数是可能的，但是这样的函数不存在吗？

点赞
用户117844
用户117844 
a = torch.randn(3,1) # 创建一个大小为(3,1)的张量a
b = torch.randn(3,1) # 创建一个大小为(3,1)的张量b
c = torch.randn(3,1) # 创建一个大小为(3,1)的张量c

d = torch.cat(a,b,1):cat(c,1) # 使用torch.cat方法将a和b在第1维上合并，并将c在第1维上与合并后的张量合并，赋值给d

print(d) # 打印d
2015-07-02 16:08:17

作者:

用户1366749

用户1366749
推荐文章