zoukankan      html  css  js  c++  java
  • torch.nonzero()和torch.index_select(),筛选张量中符合某种条件的元素

    1. torch.nonzero()的定义

    【摘自:Pytorch:torch.nonzero()函数

    2. torch.nonzero()用来筛选张量中符合某个条件的元素,得到这些元素的索引

    一维例子:

    其中的原理是,a>6得到的是一个元素为True或False的张量,如下图,True会被torch.nonzero()函数认为是非零元素,False会被认为是零元素。

    二维例子:

    3. torch.index_select()的定义和示例

    torch.index_select(input, dim, index) 函数返回的是沿着输入张量的指定维度的指定索引号进行索引的张量子集,函数参数有:

    • input(Tensor) - 需要进行索引操作的输入张量;
    • dim(int) - 需要对输入张量进行索引的维度;
    • index(LongTensor) - 包含索引号的 1D 张量;

    一维例子:

    二维例子:

    4. torch.nonzero()和torch.index_select()结合使用

    结合使用torch.nonzero()和torch.index_select(),可以选出符合某种条件的元素。下面的例子是从一维张量a中选出大于6的元素:

  • 相关阅读:
    6-2 铁轨 uva 514
    并查集基础
    周练7
    周练5
    周练4
    二分查找
    周练3
    2-7 使用不同方式进行定位.py
    2-6 使用title_contains检查页面是否正确
    启用不同浏览器.py
  • 原文地址:https://www.cnblogs.com/picassooo/p/15187175.html
Copyright © 2011-2022 走看看