Torch,如何select一个张量的子集?
Torch, how to select a subset of a tensor?
比如我有张量t = torch.rand(10,1)
。我想 select 在 {1,3,7,9}
行的张量子集并将它们连接为新张量。有什么优雅的方法吗?谢谢。
有:index
方法:
tens = torch.rand(10,1)
ind = torch.LongTensor{1,3,7,9}
subset = tens:index(1, ind)
比如我有张量t = torch.rand(10,1)
。我想 select 在 {1,3,7,9}
行的张量子集并将它们连接为新张量。有什么优雅的方法吗?谢谢。
有:index
方法:
tens = torch.rand(10,1)
ind = torch.LongTensor{1,3,7,9}
subset = tens:index(1, ind)