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

yipeiwu_com5年前Python基础

前言

在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实现修改固定模式的字符串内容操作示例

本文实例讲述了python实现修改固定模式的字符串内容操作。分享给大家供大家参考,具体如下: 说明 字符串模式是开头可能有空格,之后可能存在多个小数点,然后后面跟着一个数字,数字可能是小...

python动态视频下载器的实现方法

python动态视频下载器的实现方法

这里向大家分享一下python爬虫的一些应用,主要是用爬虫配合简单的GUI界面实现视频,音乐和小说的下载器。今天就先介绍如何实现一个动态视频下载器。 爬取电影天堂视频 首先介绍的是py...

python 环境搭建 及python-3.4.4的下载和安装过程

python 环境搭建 及python-3.4.4的下载和安装过程

第一步:下载和安装python-3.4.4amd.msi 可以去官方网站下载,也可以从网盘下载: 链接: https://pan.baidu.com/s/1hLn2y51lHiTGXCj...

关于Python 常用获取元素 Driver 总结

1、在 Windows 设置临时环境变量 cmd命令窗口 输入 path=%path%;E:\soft\python-3.5.2-embed-win32 永久配置,在系统变量下找到pat...

Django基础知识 URL路由系统详解

Django基础知识 URL路由系统详解

MVC和MTV框架 MVC Web服务器开发领域里著名的MVC模式,所谓MVC就是把Web应用分为模型(M),控制器(C)和视图(V)三层,他们之间以一种插件式的、松耦合的方式连接在...