pytorch 实现删除tensor中的指定行列
作者：qq_29007291 发布时间：2023-05-22 14:26:26
标签：pytorch,删除,tensor,行列
前言
在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]])
"""
来源：https://blog.csdn.net/qq_29007291/article/details/91042899
0
投稿猜你喜欢
- 本章的前面讨论如何使用SQL向一个表中插入数据。但是，如果你需要向一个表中添加许多条记录，使用SQL语句输入数据是很不方便的。幸运的是，My
U盘中毒了，U盘内的每个文件夹内都多了一个.lnk文件，处女座又犯了，实在不能忍，就写了个脚本把所有的.lnk文件删除了。多级目录递归删除i- 一、桥接模式桥接模式，希望能够将一个事物的两个维度分离（解耦），使其都可以独立地变化，并通过桥梁连接起来。（类）抽象部分（Abstracti
- 通过go语言实现http请求http.Postimport (?? ?"net/http"?? ?"net/u
- 加载相关库import randomimport seaborn as snsimport matplotlib.pyplot as plt
- 写出能用的代码很简单，写出好用的代码很难。好用的代码，也都会遵循一此原则，这就是设计原则，它们分别是：单一职责原则 (SRP)开闭原则 (O
- 需要在程序中使用二维数组，网上找到一种这样的用法： #创建一个宽度为3，高度为4的数组#[[0,0,0], # [0,0,0],#
- 测试代码：输出简单的ul li1.asp代码如下:<％ response.write "<ul>" r
- 前言在日常开发中 动画是必不可少的一部分 不仅能让元素直接的切换显得更加自然 同时也能极大的增强用户体验 因此 在Vue之中也提供了非常强大
- 最近社会猪可是火遍了大江南北，不蹭下热度可对不起它。见过手画的佩奇，见过用代码画的吗？没有？那就来看我大显身手。用python的turtle
- 本文实例讲述了Python实现连接MySql数据库及增删改查操作。分享给大家供大家参考，具体如下：在本文中介绍 Python3 使用PyMy
- 因为Python是自带文档，可以通过help函数来查询每一个系统函数的用法解释说明。一般来说，关键的使用方法和注意点在这个系统的文档中都说的
- 一.图像点运算概念图像点运算（Point Operation）指对于一幅输入图像，将产生一幅输出图像，输出图像的每个像素点的灰度值由输入像素
- 忽略循环引用在 .NET 5 中，如果存在循环依赖, 那么序列化的时候会抛出异常, 而在 .NET 6 中, 你可以选择忽略它。Catego
- 1、什么是线性规划线性规划（Linear programming），在线性等式或不等式约束条件下求解线性目标函数的极值问题，常用于解决资源分
- 一、array_map() 1、array_map() 函数将用户自定义函数作用到数组中的每
- 在工作和学习中如果同时传输多个文件，大的安装包，python提供了一种无线传输的方法，开启一个本地http服务器,同一局域网下可方便访问 经
- ECharts是一个纯Javascript的图表库，可以流畅的运行在PC和移动设备上，兼容当前绝大部分浏览器，底层依赖轻量级的Canvas类
- 描述int函数可以将一个指定进制的数字型字符串或者十进制数字转化为整形。语法int(object, base)名称说明备注object一个数
- 1.什么是面向对象面向对象（oop）是一种抽象的方法来理解这个世界，世间万物都可以抽象成一个对象，一切事物都是由对象构成的。应用在编程中，是
