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

相关推荐
一只小风华~29 分钟前
学习笔记:Vue Router 中的链接匹配机制与样式控制
前端·javascript·vue.js·笔记·学习·ecmascript
月临水2 小时前
Redis 学习笔记(二)
redis·笔记·学习
摇滚侠2 小时前
Spring Boot 3零基础教程,条件注解,笔记09
java·spring boot·笔记
想你依然心痛3 小时前
Spark大数据分析与实战笔记(第六章 Kafka分布式发布订阅消息系统-01)
笔记·分布式·spark
哈基鑫4 小时前
手写数字识别学习笔记
笔记·学习
润 下4 小时前
C语言——深入解析C语言指针:从基础到实践从入门到精通(三)
c语言·开发语言·经验分享·笔记·学习·程序人生·其他
润 下5 小时前
C语言——深入解析C语言指针:从基础到实践从入门到精通(二)
c语言·开发语言·经验分享·笔记·学习·程序人生
特种加菲猫5 小时前
网络协议分层:解密TCP/IP五层模型
linux·网络·笔记
小秋学嵌入式-不读研版6 小时前
C65-枚举类型
c语言·开发语言·笔记
三三木木七6 小时前
AI超级智能体学习笔记
笔记·学习