pytorch 实现删除tensor中的指定行列
作者：qq_29007291 发布时间：2023-05-22 14:26:26
标签：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]])
"""
来源：https://blog.csdn.net/qq_29007291/article/details/91042899
0
投稿猜你喜欢
- <!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML&nb
前言我们知道在这个互联网时代，评论已经在我们的生活到处可见，评论区里面的信息是一个非常有趣和有争议的地方。我们今天，就来获取某技术平台的评论- 想学习Python3，但是暂时又离不开Python2。在Windows上如何让它们共存呢？目前国内网站经常会让大家把其中一个python.e
- 大一上学期学习的内容之一，小黑屋比较好玩。1.导入函数库先导入random、time两个函数库的使用来达到随机生成人物、生成人物加载时间的目
- #@project = facepalm#@file = main#@author = Maoliang Ran#@create_time
- 任何使用yield的函数都称之为生成器，如：def count(n): while n > 0: &nb
- python开发中经常遇到报错的情况，但是warning通常并不影响程序的运行，而且有时特别讨厌，下面我们来说下如何忽略warning错误。
- 作者的blog ：http://www.planabc.net/老甘的《完全用CSS实现的中英文双语导航菜单》一文中使用“position:
- 1.echo和print的区别PHP中echo和print的功能基本相同（输出），但是两者之间还是有细微差别的。echo输出后没有返回值，但
- 本文实例讲述了thinkphp的URL路由规则与配置方法。分享给大家供大家参考。具体分析如下：一、URL规则1、默认是区分大小写的2、如果我
- django创建自定义模板处理器:一、需求来源：在django开发中，页面是通过template（模板）进行渲染的，对于一些数据，可以通过{
- 转:coolcode.cn通常情况下，我们的网页要指定一个编码字符集，如 GB2312、UTF-8、ISO-8859-1
- 使用jmail组件发送邮件：Function JMail(Send_From,Send_To,Send_Subject,Send_
- 对象Javascript 根本上是和对象相关的。数组是对象。函数是对象。对象是对象。那什么是对象呢？对象是名-值对的集合。名是字符串，值可以
- python 列表和链表的区别python 中的 list 并不是我们传统意义上的列表，传统列表——通常也叫作链表(linked list)
- xhtml+css页面制作过程中问题的解决方案，说是解决方案应该有点过了，充其量只不过是给刚刚开始学标准页面制作的朋友们的一些小建议，如果讲
- 这里假定Postgresql数据库已经装好。首先安装依赖的包$ sudo yum install python-devel postgres
- # 半夜撸代码 正在一顿操作猛如虎的时候,发现删了其中一张表的某条记录,结果发现其他表跟这个字段的关联的也都被删除,我已经写了d
- Request.ServerVariables里Server_Name与Http_Host之间有什么区别呢？如果不仔细看，您应该看不出它们之
- 在一些网页应用中，就比如在投票系统中，当我们进行的是多项投票时，我们要求用户最多只能选择几项进行投票，这也是就是说选择复选框的个数最多几个.
