【问题标题】:Torch, how to select a subset of a tensor?Torch,如何选择张量的子集?
【发布时间】:2016-07-05 12:45:35
【问题描述】:

例如,我有一个张量t = torch.rand(10,1)。我想选择{1,3,7,9} 行的张量子集并将它们连接为新张量。有什么优雅的方法吗?谢谢。

【问题讨论】:

    标签: lua torch


    【解决方案1】:

    :index方法:

    tens = torch.rand(10,1)
    ind = torch.LongTensor{1,3,7,9}
    subset = tens:index(1, ind)
    

    【讨论】:

      猜你喜欢
      • 2015-09-27
      • 2019-11-12
      • 2018-01-05
      • 1970-01-01
      • 2016-05-02
      • 2021-10-07
      • 2016-07-02
      • 2016-03-15
      • 2017-09-11
      相关资源
      最近更新 更多