pytorch 实现删除tensor中的指定行列


Posted in Python onJanuary 13, 2020

前言

在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 相关文章推荐
python结合opencv实现人脸检测与跟踪
Jun 08 Python
Python中扩展包的安装方法详解
Jun 14 Python
Python标准库shutil用法实例详解
Aug 13 Python
深入理解Python中的 __new__ 和 __init__及区别介绍
Sep 17 Python
python整小时 整天时间戳获取算法示例
Feb 20 Python
python实现栅栏加解密 支持密钥加密
Mar 20 Python
python3读取图片并灰度化图片的四种方法(OpenCV、PIL.Image、TensorFlow方法)总结
Jul 04 Python
python数据预处理之数据标准化的几种处理方式
Jul 17 Python
Python Gitlab Api 使用方法
Aug 28 Python
TensorFLow 数学运算的示例代码
Apr 21 Python
利用matplotlib为图片上添加触发事件进行交互
Apr 23 Python
python openCV自制绘画板
Oct 27 Python
python基于event实现线程间通信控制
Jan 13 #Python
pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
Jan 13 #Python
python、PyTorch图像读取与numpy转换实例
Jan 13 #Python
pytorch 彩色图像转灰度图像实例
Jan 13 #Python
Ranorex通过Python将报告发送到邮箱的方法
Jan 12 #Python
python opencv实现信用卡的数字识别
Jan 12 #Python
Python 实现递归法解决迷宫问题的示例代码
Jan 12 #Python
You might like
php检查日期函数checkdate用法实例
2015/03/19 PHP
PHP safe_mode开启对于PHP系统函数有什么影响
2020/11/10 PHP
PHP7新增函数
2021/03/09 PHP
跨浏览器的设置innerHTML方法
2006/09/18 Javascript
js实现的网站首页随机公告随机公告
2007/03/14 Javascript
Firefox和IE浏览器兼容JS脚本写法小结
2008/07/07 Javascript
Extjs4中Form的使用之本地hiddenfield
2013/11/26 Javascript
javascript实现鼠标移到Image上方时显示文字效果的方法
2015/08/07 Javascript
jQuery Easyui实现左右布局
2016/01/26 Javascript
nodejs利用http模块实现银行卡所属银行查询和骚扰电话验证示例
2016/12/30 NodeJs
ES6入门教程之Iterator与for...of循环详解
2017/05/17 Javascript
在vue.js中使用JSZip实现在前端解压文件的方法
2018/09/05 Javascript
浅谈webpack4 图片处理汇总
2018/09/12 Javascript
vue实现抖音时间转盘
2019/09/08 Javascript
[01:18]PWL开团时刻DAY10——一拳超人
2020/11/11 DOTA
简单介绍Python的轻便web框架Bottle
2015/04/08 Python
详解Python的Django框架中的模版继承
2015/07/16 Python
使用C#配合ArcGIS Engine进行地理信息系统开发
2016/02/19 Python
python实现单线程多任务非阻塞TCP服务端
2017/06/13 Python
python 垃圾收集机制的实例详解
2017/08/20 Python
在python中利用最小二乘拟合二次抛物线函数的方法
2018/12/29 Python
Python提取支付宝和微信支付二维码的示例代码
2019/02/15 Python
Python 占位符的使用方法详解
2019/07/10 Python
pytorch逐元素比较tensor大小实例
2020/01/03 Python
Python3将ipa包中的文件按大小排序
2020/04/17 Python
Python之字符串的遍历的4种方式
2020/12/08 Python
java字符串格式化输出实例讲解
2021/01/06 Python
HTML5离线缓存在tomcat下部署可实现图片flash等离线浏览
2012/12/13 HTML / CSS
递归实现回文判断(如:abcdedbca就是回文,判断一个面试者对递归理解的简单程序)
2013/04/28 面试题
售前工程师职业生涯规划
2014/03/02 职场文书
作风年建设汇报材料
2014/08/14 职场文书
十八大宣传标语
2014/10/09 职场文书
政协会议宣传标语
2014/10/09 职场文书
税务会计岗位职责
2015/04/02 职场文书
Golang 并发下的问题定位及解决方案
2022/03/16 Golang
Python 读取千万级数据自动写入 MySQL 数据库
2022/06/28 Python