pytorch 实现删除tensor中的指定行列
2020-02-13 13:13
2699 查看
前言
在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中的指定行列就是小编分享给大家的全部内容了,希望能给大家一个参考
您可能感兴趣的文章:
相关文章推荐
- pytorch 实现tensor与numpy数组转换
- Pytorch使用MNIST数据集实现CGAN和生成指定的数字方式
- pytorch 实现张量tensor,图片,CPU,GPU,数组等的转换
- Pytorch使用MNIST数据集实现CGAN和生成指定的数字
- tensorboard、jupyter notebook、Pytorch visdom指定端口
- 原生js实现查找/添加/删除/指定元素的class
- Asp.net+Extjs实现对GridPanel分页、行内编辑、增加、删除、弹出新窗体增加、数据验证、编辑指定列、数据格式转换等功能
- 使用存储过程实现删除指定列名上存在的索引
- php实现删除指定目录下相关文件的方法
- 删除当前文件夹下指定类型的文件 使用批处理实现
- 修改simple_html_dom源码 实现删除指定html标签
- spring+quartz实现定时任务(每天零点删除指定文件夹中缓存文件)
- python利用os和getopt实现删除指定文件
- BAT 实现删除指定天数之前的文件
- pytorch 获取tensor维度信息示例
- windows api 实现删除指定目录下的所有文件(包括子文件夹下的所有文件)
- 编程实现删除字符串中所有指定的字符
- pytorch sampler对数据进行采样的实现
- Pytorch使用MNIST数据集实现基础GAN和DCGAN详解
- pytorch crayon tensorboard