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

相关推荐
1104.北光c°5 分钟前
我理解的Leaf号段模式:美团分布式ID生成系统
java·开发语言·笔记·分布式·github·leaf
就叫飞六吧35 分钟前
两步实现支付宝沙箱
笔记
Stream_Silver36 分钟前
【系统架构设计师】第一章 计算机硬件 1.1 计算机硬件组成
笔记·硬件架构
山川行40 分钟前
Git学习笔记:Git进阶操作
笔记·git·vscode·学习·编辑器·visual studio code
CDN3601 小时前
运维笔记|360CDN高防服务器部署教程,抗D+源站防护一站式配置
运维·服务器·笔记
左左右右左右摇晃1 小时前
Java Object 类笔记
java·笔记
橘bird1 小时前
LangChain1.2 学习笔记(自用)(未完结)
笔记·python·学习·langchain
智者知已应修善业1 小时前
【任何一个自然数m的立方均可写成m个连续奇数之和】2024-10-17
c语言·数据结构·c++·经验分享·笔记·算法
吃着火锅x唱着歌1 小时前
PHP7内核剖析 学习笔记 第十章 扩展开发(3)
java·笔记·学习