本文介绍了火炬，如何选择张量的子集?的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
例如，我有一个张量t = torch.rand(10,1).我想在{1,3,7,9}行中选择张量子集，并将它们连接为新张量.有什么优雅的方法吗?谢谢.
For example, I have a tensor t = torch.rand(10,1). And I want to select the tensor subset of at the rows of {1,3,7,9} and concatenate them as new tensor. Is there any elegant way? Thank you.
推荐答案
有 :index 方法:
tens = torch.rand(10,1)
ind = torch.LongTensor{1,3,7,9}
subset = tens:index(1, ind)
这篇关于火炬，如何选择张量的子集?的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！