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

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博客

相关推荐
Roselind_Yi4 分钟前
云计算实验实操|Hadoop伪分布式部署+MapReduce编程实践(超详细图文版)
大数据·hadoop·经验分享·笔记·分布式·数据挖掘·云计算
safedebug18 分钟前
此服务器的证书无效 您可能正在连接到一个伪装
笔记·学习
做cv的小昊23 分钟前
【TJU】应用统计学——第六周作业(3.3 两个正态总体参数的假设检验、3.4 非正态总体参数的假设检验、4.1 一元线性回归分析)
笔记·算法·数学建模·矩阵·回归·线性回归·学习方法
晨晖225 分钟前
linux笔记6
linux·运维·笔记
古方路杰出青年28 分钟前
学习笔记1:Python FastAPI极简后端API示例解析
笔记·后端·python·学习·fastapi
羊群智妍31 分钟前
2026年GEO监测工具大全|免费AI搜索优化直接用
笔记
江湖人称小鱼哥11 小时前
Obsidian-Graphify-让你的笔记库自己长出知识图谱
笔记·知识图谱·obsidian·claude code·graphify·卡帕西
苦 涩13 小时前
考研408笔记之计算机网络(三)——数据链路层
笔记·计算机网络·考研408
三品吉他手会点灯13 小时前
STM32F103 学习笔记-21-串口通信(第4节)—串口发送和接收代码讲解(中)
笔记·stm32·单片机·嵌入式硬件·学习
雾岛听蓝14 小时前
Qt操作指南:窗口组成与菜单栏
开发语言·经验分享·笔记·qt