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自动化部署工具Fabric的简单上手指南
Apr 19 Python
Python实现翻转数组功能示例
Jan 12 Python
python写一个md5解密器示例
Feb 23 Python
python将每个单词按空格分开并保存到文件中
Mar 19 Python
selenium+python实现1688网站验证码图片的截取功能
Aug 14 Python
python查找指定文件夹下所有文件并按修改时间倒序排列的方法
Oct 21 Python
python飞机大战pygame游戏背景设计详解
Dec 17 Python
python nohup 实现远程运行不宕机操作
Apr 16 Python
python和JavaScript哪个容易上手
Jun 23 Python
python 绘制正态曲线的示例
Sep 24 Python
scrapy在python爬虫中搭建出错的解决方法
Nov 22 Python
python神经网络学习 使用Keras进行简单分类
May 04 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依赖注入原理与用法分析
2018/08/21 PHP
启用OPCache提高PHP程序性能的方法
2019/03/21 PHP
原始的js代码和jquery对比体会
2013/09/10 Javascript
JS小功能(checkbox实现全选和全取消)实例代码
2013/11/28 Javascript
Chrome扩展页面动态绑定JS事件提示错误
2014/02/11 Javascript
详解JavaScript中的every()方法
2015/06/08 Javascript
在ASP.NET MVC项目中使用RequireJS库的用法示例
2016/02/15 Javascript
jQuery Mobile页面返回不需要重新get
2016/04/26 Javascript
基于JQuery的$.ajax方法进行异步请求导致页面闪烁的解决办法
2016/05/10 Javascript
js图片上传前预览功能(兼容所有浏览器)
2016/08/24 Javascript
javascript checkbox/radio onchange不能兼容ie8处理办法
2017/06/13 Javascript
详解基于vue的移动web app页面缓存解决方案
2017/08/03 Javascript
JS实现问卷星自动填问卷脚本并在两秒自动提交功能
2020/06/17 Javascript
Angular2.0实现modal对话框的方法示例
2018/02/18 Javascript
父组件中vuex方法更新state子组件不能及时更新并渲染的完美解决方法
2018/04/25 Javascript
Vue数据双向绑定的深入探究
2018/11/27 Javascript
layui数据表格 table.render 报错的解决方法
2019/09/29 Javascript
Python中struct模块对字节流/二进制流的操作教程
2017/01/21 Python
Python多进程并发与多线程并发编程实例总结
2018/02/08 Python
详解Appium+Python之生成html测试报告
2019/01/04 Python
python cumsum函数的具体使用
2019/07/29 Python
Python调用REST API接口的几种方式汇总
2020/10/19 Python
美国最受欢迎的度假租赁网站:VRBO
2016/08/02 全球购物
鞋子女王塔玛拉·梅隆同名奢侈品牌:Tamara Mellon
2017/11/22 全球购物
芬兰设计商店美国:Finnish Design Shop US
2019/03/25 全球购物
简单说下OSPF的操作过程
2014/08/13 面试题
深圳茁壮笔试题
2015/05/28 面试题
自我鉴定书范文
2013/10/02 职场文书
技术总监的工作职责
2013/11/13 职场文书
学习新党章思想汇报
2014/01/09 职场文书
国旗下的演讲稿
2014/05/08 职场文书
讲文明知礼仪演讲稿
2014/09/13 职场文书
2014年纪检监察工作总结
2014/11/11 职场文书
社区志愿服务活动感想
2015/08/07 职场文书
技术入股协议书
2016/03/22 职场文书
实战 快速定位MySQL的慢SQL
2022/03/22 MySQL