常用函数备忘-nonzero()
pytorch吧
全部回复
仅看楼主
level 11
sealed_ss
楼主
今天遇到了一个没见过的函数，nonzero()。这个函数是用来检测张量中不为0的元素的index，默认情况下返回的是二维张量，张量中存储的就是不为0元素的index。
2023年09月18日 10点09分
1
level 1
韫玉1596
有测试过速度嘛和(tensor.abs()>1e-7)这种方式谁的速度更快
2023年09月18日 15点09分
2
sealed_ss
看4楼
2023年09月19日 10点09分
level 3
🌕、✨
2023年09月18日 23点09分
3
level 11
sealed_ss
楼主
2023年09月19日 10点09分
4
sealed_ss
看起来是nonzero方式更快
2023年09月19日 10点09分
level 1
韫玉1596
3倍的区别，看样子也还能接受
。不是差特别多
2023年09月19日 10点09分
5
sealed_ss
2023年09月20日 10点09分
1
![[乖]](/static/emoticons/u4e56.png)
今天遇到了一个没见过的函数，nonzero()。这个函数是用来检测张量中不为0的元素的index，默认情况下返回的是二维张量，张量中存储的就是不为0元素的index。![[小乖]](/static/emoticons/u5c0fu4e56.png)
有测试过速度嘛和(tensor.abs()>1e-7)这种方式谁的速度更快
![[怒]](/static/emoticons/u6012.png)
看4楼
![[真棒]](/static/emoticons/u771fu68d2.png)
![[啊]](/static/emoticons/u554a.png)
看起来是nonzero方式更快![[阴险]](/static/emoticons/u9634u9669.png)