技术栈

笔记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章 备忘录模式
相关推荐
浩瀚地学
2 小时前
【Java】JDK8的一些新特性
java·开发语言·经验分享·笔记·学习
JeffDingAI
2 小时前
【Datawhale学习笔记】深入大模型架构
笔记·学习
a不是橘子
2 小时前
03在Ubuntu中验证PV操作
笔记·ubuntu·操作系统·虚拟机·os·pv操作
tangyal
2 小时前
渗透笔记1
笔记·网络安全·渗透
fanged
3 小时前
STM32(5)--HAL1(TODO)
笔记
grd4
5 小时前
RN for OpenHarmony 小工具 App 实战:屏幕尺子实现
笔记·harmonyos
丝斯2011
6 小时前
AI学习笔记整理(50)——大模型中的Graph RAG
人工智能·笔记·学习
小+不通文墨
6 小时前
“超声波测量声速”实验报告
经验分享·笔记·学习·学习方法
lihao lihao
7 小时前
平衡二叉树
笔记
JavaLearnerZGQ
8 小时前
我的Redis笔记2【分布式缓存】
redis·笔记·缓存
热门推荐
01GitHub 镜像站点022025 Telegram 最新免费社工库机器人(LetsTG可[特殊字符])搭建指南(含 Python 脚本)03OpenCode 入门教程:介绍 · 安装 · 配置第三方 API (如 Claude)04UV安装并设置国内源05Linux下V2Ray安装配置指南06AI 规范驱动开发“三剑客”深度对比:Spec-Kit、Kiro 与 OpenSpec 实战指南07安娜的档案(Anna’s Archive) 镜像网站/国内最新可访问入口(持续更新)08Claude Code Skills 实用使用手册09Labelme从安装到标注:零基础完整指南10BongoCat - 跨平台键盘猫动画工具