Pytorch基础(21)-- torch.repeat_interleave()方法

分享一下自己目前在维护的Github项目,由于本人博士阶段接触了一个全新的研究方向-----使用机器学习、强化学习、深度学习等方法解决组合优化问题,维护这个项目的目的:
(1)记录自己阅读过的paper,同时分享一下对某些paper的见解;
(2)方便其他的初学者或研究生查阅最新的paper;
⭐️ 项目链接:Welcom to COPZoo

一、前言

嗯……,又是奇奇怪怪的方法,果然,你能想到的功能,大佬们已经实现了。今天要分享的内容是torch.repeat_interleave() 方法。话不多说,Let’s go!

二、方法解析

官方对repeat_interleave方法的介绍及其简单,介绍的越简单,对于使用者来说越难以理解。官方解释:重复一个tensor中元素。嗯……确实简单!

参数

  • input:白话就是你要操作的tensor
  • repeat:每个元素要重复的次数,可以是int型;也可以是一个Tensor,但要注意的是,这个Tensor进行广播后,要符合给定维度的shape。
  • dim:可选参数,你想要沿着哪个维度进行重复。值得注意的是:如果不设置,默认首先将input展开为1维tensor,然后再进行重复。
    output_size:可选参数,一般用不到。这里没研究。

在这里插入图片描述

三、案例分析

import numpy as np
import torch

3.1 案例1 - repeats为default

t = torch.tensor([[1,2,3],[4,5,6]])
t

在这里插入图片描述

t1 = torch.repeat_interleave(input=t, repeats=2)
t1

在这里插入图片描述

3.2 案例2 - repeats为int型

repeats=0
t2 = torch.repeat_interleave(input=t, repeats=2, dim=0)
t2

在这里插入图片描述

repeats=1
t3 = torch.repeat_interleave(input=t, repeats=2, dim=1)
t3

在这里插入图片描述

3.3 案例3 - repeats为tensor型

t4 = torch.repeat_interleave(input=t, repeats=torch.tensor([2,3]), dim=0)
t4

在这里插入图片描述

t5 = torch.repeat_interleave(input=t, repeats=torch.tensor([2,3,1]), dim=1)
t5

在这里插入图片描述

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/535111.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

浅谈如何自我实现一个消息队列服务器(1)——需求分析

文章目录 一、什么是消息队列?二、当下主流的消息队列(MQ)三、自我实现一个消息队列服务器的前期准备——需求分析3.1 核心概念3.2 broker server 核心概念3.2.1、虚拟主机(Virtual Host)3.2.2、交换机(Exchange)3.2.2…

外包干了5天,技术明显退步。。。。。

先说一下自己的情况,本科生,19年通过校招进入南京某软件公司,干了接近2年的功能测试,今年年初,感觉自己不能够在这样下去了,长时间呆在一个舒适的环境会让一个人堕落!而我已经在一个企业干了2年的功能测试&…

BEC报考公告 ,柯桥成人学商务英语,商务英语口语学校

BEC报考公告 报名时间 2024年3月12日10:00——2023年3月20日10:00 注册个人信息、上传电子照片并支付考试费用 考试时间 BEC初级:5月12日 BEC中级:5月25日 BEC高级:5月18日 笔试及口试具体时间以准考证为准 报名费用 初级:…

C++矢量运算与java矢量运算

矢量运算 概述: 矢量运算是一种基于向量的数学运算,它遵循特定的法则。以下是矢量运算的一些基本原理: 矢量加法:可以使用平行四边形法则或三角形法则来执行。当两个矢量相加时,可以将它们的起点放在同一个点上&…

朴素贝叶斯算法基础——案例:对新闻进行分类

贝叶斯公式 朴素:假设特征与特征之间相互独立 朴素贝叶斯算法:朴素贝叶斯 应用场景:文本分类(单词作为特征) 拉普拉斯平滑系数 Ni:F1词在C类别所有文档中出现的次数 N:所属类别C下的文档所…

华为车控面试前后

个人经历: 秋招未接受其他公司offer,all in华子。 ->秋招失败0 offer 年前被车bu捞后入池开始审批。 ->等待超过1个月,陷入煎熬。 ->终于等到意向书。 分享时间线: 10月 笔试和3面入池2012 1月 收到车bu捞人电话解…

JVM是如何运行的

JVM(Java Virtual Machine,Java虚拟机)是 Java 程序的运行环境,它负责将 Java 字节码翻译成机器代码并执行。也就是说 Java 代码之所以能够运行,主要是依靠 JVM 来实现的。 JVM 整体的大概执行流程是这样的&#xff1…

尚硅谷SpringBoot3笔记

推荐课程:03.快速入门-示例Demo_哔哩哔哩_bilibili 官方笔记:SpringBoot3全栈指南 (yuque.com) 目录 01 -- 示例demo 1. 在新建项目创建一个Maven 模块 2. 引入 spring-boot-starter-parent 和 spring-boot-starter-web 依赖 3. 指示 Spring Boot 启…

Spring多线程事务处理

一、背景 本文主要介绍了spring多线程事务的解决方案,心急的小伙伴可以跳过上面的理论介绍分析部分直接看最终解决方案。 在我们日常的业务活动中,经常会出现大规模的修改插入操作,比如在3.0的活动赛事创建,涉及到十几张表的插入…

doris安装(docker方式)

背景 doris有两个进程 fe,处理用户请求,查询,元数据管理,节点管理be,数据存储,查询计划执行 架构图如下: 参考:https://doris.apache.org/zh-CN/docs/get-starting/what-is-apache-doris 1、定义docker-compose文件 version: 3 services:docker-fe:image: "apac…

提升用户体验,Xinstall让微信内打开第三方APP更顺畅

在移动互联网时代,APP推广的重要性不言而喻。然而,推广过程中我们经常会遇到各种问题,比如用户在微信内无法直接打开第三方APP,导致推广效果大打折扣。那么,如何在微信内实现无缝跳转,提高推广效率呢&#…

Python使用错误总结

【1】cannot import name ‘ParameterSource’ from ‘click.core’ 其根本原因在于是black模块,其模块版本可能过时,升级black模块版本即可: pip install black --upgrade【2】partially initialized module ‘charset_normalizer’ has n…