【发布时间】:2016-07-05 12:45:35
【问题描述】:
例如,我有一个张量t = torch.rand(10,1)。我想选择{1,3,7,9} 行的张量子集并将它们连接为新张量。有什么优雅的方法吗?谢谢。
【问题讨论】:
例如,我有一个张量t = torch.rand(10,1)。我想选择{1,3,7,9} 行的张量子集并将它们连接为新张量。有什么优雅的方法吗?谢谢。
【问题讨论】:
有:index方法:
tens = torch.rand(10,1)
ind = torch.LongTensor{1,3,7,9}
subset = tens:index(1, ind)
【讨论】: