pytorch 实现删除tensor中的指定行列
更新时间：2020年01月13日 09:18:42 作者：qq_29007291
今天小编就为大家分享一篇pytorch 实现删除tensor中的指定行列，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
前言
在pytorch中， 想删除tensor中的指定行列，原本以为有个函数或者直接把某一行赋值为[]就可以，结果发现没这么简单，因此用了一个曲线救国方法，希望如果有更直接的方法，请大家指出。
code
本质上是利用mask删除了指定行，然后重新指向。
a = torch.rand(4, 2)
print(a)
idx = 1
a = a[torch.arange(a.size(0))!=1]
print(a)
"""
tensor([[2.7775e-01, 3.7430e-01],
[9.0373e-01, 8.1220e-02],
[9.8638e-01, 8.6293e-01],
[9.8139e-04, 9.8460e-02]])
tensor([[2.7775e-01, 3.7430e-01],
[9.8638e-01, 8.6293e-01],
[9.8139e-04, 9.8460e-02]])
"""
以上这篇pytorch 实现删除tensor中的指定行列就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
下面小编就为大家带来一篇简单谈谈python中的语句和语法。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-08-08
这篇文章主要介绍了用Python实现一个简单的线程池,通过这个小程序可以帮助更好地理解Python中线程的运行机制,需要的朋友可以参考下2015-04-04
这篇文章主要为大家详细介绍了Python实现LRU缓存置换算法，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-05-05
本篇文章给大家分享了用python脚本写出作为Windows服务启动功能，对此有需求的朋友跟着小编一起学习下。2018-02-02
这篇文章主要介绍了【python】str与json类型转换，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-04-04
这篇文章主要介绍了使用Python打造一款间谍程序,本文通过实例代码给大家介绍的非常详细，具有一定的参考借鉴价值，需要的朋友可以参考下2020-02-02
这篇文章主要给大家介绍了关于Python求最小公倍数4种方法的相关资料,最小公倍数不可以像最大公约数那样直接利用辗转相除法求出,但可以借助辗转相除法求得的最大公约数来求最小公倍数,需要的朋友可以参考下2023-10-10
这篇文章主要介绍了Python中协程用法代码详解，分享了相关代码示例，小编觉得还是挺不错的，具有一定借鉴价值，需要的朋友可以参考下2018-02-02
解决ImportError:cannot import name ‘Flatten‘&nb
这篇文章主要介绍了解决ImportError:cannot import name ‘Flatten‘ from ‘torch.nn‘问题，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-06-06
这篇文章主要介绍了Python自动化测试selenium指定截图文件名方法,Selenium 支持 Web 浏览器的自动化,它提供一套测试函数,用于支持 Web 自动化测试,下文基于python实现指定截图文件名方法,需要的小伙伴可以参考一下2025-02-02
最新评论