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基于smtplib实现异步发送邮件服务

基于smtplib包制作而成,但在实践中发现一个不知道算不算是smtplib留的一个坑,在网络断开的情况下发送邮件时会抛出一个socket.gaierror的异常,但是smtplib中并...

python实现图片插入文字

python实现图片插入文字

本文实例为大家分享了python图片插入文字的具体代码,供大家参考,具体内容如下 问题 如何在图片中插入大量文字并且自动换行 效果 原始图 效果图 注明 若需要写入中文请使用中文字体...

根据tensor的名字获取变量的值方式

需求: 有时候使用slim这种封装好的工具,或者是在做滑动平均时,系统会帮你自动建立一些变量,但是这些变量只有名字,而没有显式的变量名,所以这个时候我们需要使用那个名字来获取其对应的值。...

详解JavaScript编程中的window与window.screen对象

Window 对象 所有浏览器都支持 window 对象。它表示浏览器窗口。 所有 JavaScript 全局对象、函数以及变量均自动成为 window 对象的成员。 全局变量是 win...

opencv转换颜色空间更改图片背景

本文实例为大家分享了opencv转换颜色空间更改图片背景的具体代码,供大家参考,具体内容如下 思路: 1、将BGR转换为HSV颜色空间 2、设置掩模 3、位运算 这里以更改摩托罗拉l...