pytorch索引查找 index_select的例子
作者：ShellCollector 发布时间：2023-09-12 20:45:05
标签：pytorch,索引,查找,index,select
index_select
anchor_w = self.FloatTensor(self.scaled_anchors).index_select(1, self.LongTensor([0]))
参数说明：index_select(x, 1, indices)
1代表维度1，即列，indices是筛选的索引序号。
例子：
import torch
x = torch.linspace(1, 12, steps=12).view(3,4)
print(x)
indices = torch.LongTensor([0, 2])
y = torch.index_select(x, 0, indices)
print(y)
z = torch.index_select(x, 1, indices)
print(z)
z = torch.index_select(y, 1, indices)
print(z)
结果：
tensor([[ 1., 2., 3., 4.],
[ 5., 6., 7., 8.],
[ 9., 10., 11., 12.]])
tensor([[ 1., 2., 3., 4.],
[ 9., 10., 11., 12.]])
tensor([[ 1., 3.],
[ 5., 7.],
[ 9., 11.]])
tensor([[ 1., 3.],
[ 9., 11.]])
来源：https://blog.csdn.net/jacke121/article/details/83044660
0
投稿猜你喜欢
- 1、矩阵加法使用a = np.random.random((3,3))b = np.random.randint(0,9,(3,3))ad
- pandas.DataFrame为每一列保存一个数据类型dtype。要仅提取（选择）特定数据类型为dtype的列，请使用pandas.Dat
- 如何做一个全面的探测器？ 我们也可以做一个功能类似的探测器，见下：<Script lan
- 如下所示：# -*- coding:utf-8 -*-import sysreload(sys)sys.setdefaultencoding
条形图（bar chart），也称为柱状图，是一种以长方形的长度为变量的统计图表，长方形的长度与它所对应的变量数值呈一定比例。1. 竖放条形- 本文实例讲述了php获取客户端IP及URL的方法。分享给大家供大家参考，具体如下：function getonlineip(){//获取用户
- 首先，‘and’、‘or’和‘not&a
- 来自巴西的设计师Roger Oddone的作品，通过此作品你可以了解到logo的设计的一些思路。
- 概述这是一个简单的python爬虫程序，仅用作技术学习与交流，主要是通过一个简单的实际案例来对网络爬虫有个基础的认识。什么是网络爬虫简单的讲
- 今天遇到一个问题，就是用pycharm运行python程序，老是会出现Python.exe已停止的对话框。后来我到处在网上搜原因，网上给出的
- vue做移动端经常碰到弹窗的需求, 这里写一个功能简单的vue弹窗popup.vue<template> <div cla
- python版本为python3.51.要求1）输入用户名密码2）认证成功后显示欢迎信息3）输错三次后锁定2.需求分析1）用户信息存储在文件
- 现象：已知，连接的WIFI网络需要通过代理服务器才能连接外网，按照正常的程序无法发送邮件，而直连一个没有代理的网络【如自己的wifi热点】，
- 想要根据django中的模型和配置生成SQL语句，需要先进行一定的设置：首先需要在你的app文件夹中进入setting.py文件，里面有一个
- 1. 导入库import numpy as np #矩阵运算import matplotlib.pyplot as plt #可视化impo
- 一、根据条件在序列中筛选数据假设有一个数字列表 data, 过滤列表中的负数data = [1, 2, 3, 4, -5]# 使用列表推导式
- 本文主要介绍下如何使用第三方库。1. 理解第三方库Python相当于一个手机，第三方库相当于手机里各种各样的APP。当我们想搭建网站时，可以
- 随着网页制作热潮的兴起，Dreamweaver 4.0强大的功能深受众多网页制作者的喜爱。特别是Dreamweaver 4.0中有许多第三方
- 本文实例为大家分享了Python实现五子棋游戏的具体代码，供大家参考，具体内容如下了解游戏的规则是我们首先需要做的事情，如果不知晓规则，那么
- RDPY 是基于 Twisted Python 实现的微软 RDP 远程桌面协议。RDPY 提供了如下 RDP 和 VNC 支持：●RDP
