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

相关推荐
三品吉他手会点灯14 小时前
C语言学习笔记 - 12.C语言简介 - 一元二次方程详解
c语言·笔记·学习
Cathy Bryant15 小时前
微分几何:度规(度量)metric
笔记·线性代数·矩阵·高等数学·物理
fengxin_rou16 小时前
JVM 核心笔记:对象创建、生命周期与类加载器详解
java·jvm·笔记
weixin_7042660516 小时前
Spring Cloud Gateway 完整版笔记
笔记
qeen8717 小时前
【算法笔记】二分查找与二分答案
c语言·c++·笔记·学习·算法·二分
脆皮炸鸡75517 小时前
进程的程序替换
linux·经验分享·笔记·vim·学习方法
萌新小码农‍17 小时前
机器学习概述 学习笔记day2
笔记·学习·机器学习
daanpdf17 小时前
大一《中国近代史纲要》题库及答案PDF知识点整理笔记
笔记·pdf
曦月逸霜17 小时前
区块链技术与应用学习笔记(持续更新中)
笔记·学习·区块链
中屹指纹浏览器19 小时前
跨终端跨平台指纹浏览器的环境一致性技术架构与落地实践
经验分享·笔记