技术栈

笔记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章 备忘录模式
相关推荐
yyyyy_abc
8 小时前
ceph学习笔记
笔记·ceph·学习
晓梦林
8 小时前
ximai靶场学习笔记
android·笔记·学习
IT1995
8 小时前
Dify笔记-创建自定义工具连接到自建的OpenAPI Schema服务
笔记
一口吃俩胖子
9 小时前
【脉宽调制DCDC功率变换学习笔记021】时域性能准则
笔记·学习
starvapour
9 小时前
SSH笔记
笔记·ssh
三品吉他手会点灯
10 小时前
C语言学习笔记 - 40.数据类型 - scanf函数的编程规范与非法输入处理
c语言·开发语言·笔记·学习
大G的笔记本
11 小时前
synchronized 的原理(简单版)
笔记
daad777
12 小时前
记录一个希尔伯特曲线笔记
笔记
江屿风
12 小时前
C++OJ题经验总结(竞赛)1
开发语言·c++·笔记·算法
wangcheng303
13 小时前
AI+制造正悄然改变工业生产
笔记
热门推荐
01GitHub 镜像站点02Codex 接入 DeepSeek API 完整配置文档03CC-Switch & Claude 基于 Linux 服务器安装使用指南04用了半年 OpenRouter,我换到了 Ofox.ai — 两个 AI API 聚合平台的真实对比05几个好用的ip纯净度检测网站06裂开!ChatGPT 居然开始要手机号验证,附详细解决方法07Gemini大升级、AI眼镜首发、Android XR亮相,13天后见分晓08【踩坑记录 | 第一篇】微软商店无法使用时,如何手动安装 OpenAI Codex?附`.msix`文件系统错误解决方法09【AI】2026 年具身智能模型和世界模型总结10codex app每次打开重连5次Reconnecting问题解决