技术栈

笔记68:Pytorch中repeat函数的用法

恨晨光熹微2023-12-10 17:33

repeat 相当于一个broadcasting的机制

repeat(*sizes)

沿着指定的维度重复tensor。不同与expand(),本函数复制的是tensor中的数据。

python 复制代码
import torch
import torch.nn.functional as F
import numpy as np
a = torch.Tensor(128,1,512)
B = a.repeat(1,5,1)
print(B.shape)
python 复制代码
torch.Size([128, 5, 512])

转自:pytorch repeat的用法-CSDN博客

上一篇:SpringBoot实战项目:蚂蚁爱购(从零开发)
下一篇:《C++新经典设计模式》之第18章 备忘录模式
相关推荐
特种加菲猫
1 小时前
指尖上的魔法:优雅高效的Linux命令手册
linux·笔记
wuxuanok
1 小时前
Web后端开发-分层解耦
java·笔记·后端·学习
wuxuanok
2 小时前
Web后端开发-请求响应
java·开发语言·笔记·学习
诗句藏于尽头
3 小时前
内网使用rustdesk搭建远程桌面详细版
笔记
蜡笔小电芯
3 小时前
【C语言】指针与回调机制学习笔记
c语言·笔记·学习
丰锋ff
3 小时前
瑞斯拜考研词汇课笔记
笔记
DKPT
5 小时前
Java享元模式实现方式与应用场景分析
java·笔记·学习·设计模式·享元模式
KoiHeng
8 小时前
操作系统简要知识
linux·笔记
巴伦是只猫
9 小时前
【机器学习笔记Ⅰ】11 多项式回归
笔记·机器学习·回归
DKPT
12 小时前
Java桥接模式实现方式与测试方法
java·笔记·学习·设计模式·桥接模式
热门推荐
01Java学习第十五部分——MyBatis02集群聊天服务器---MySQL数据库的建立03Coze扣子平台完整体验和实践(附国内和国际版对比)04基于odoo17的设计模式详解---装饰模式05使用Ruby接入实时行情API教程06扣子(coze)实战|我用扣子搭建了一个自动分析小红薯笔记内容的AI应用|详细步骤拆解07Everything文件检索工具 几秒检索几百G的文件08基于odoo17的设计模式详解---单例模式09DeepSeek各版本说明与优缺点分析10【无标题】