Torch，如何选择张量的子集？

2016-3-18 16:45:15
收藏：0
阅读：48
评论：1

例如，我有一个张量 t = torch.rand(10,1)。我想选择在行上的子集 {1,3,7,9} 并将它们连接成新的张量。有没有一种优雅的方法？谢谢。

点赞
用户4850610
用户4850610

有一个 :index 方法：

tens = torch.rand(10,1)
ind = torch.LongTensor{1,3,7,9}
subset = tens:index(1, ind)
2016-03-18 21:00:41

作者:

用户3449534

用户3449534
推荐文章