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

相关推荐
喝醉酒的小白2 小时前
Elasticsearch 故障分析笔记:Pending Tasks 堆积与 Alias 风暴
笔记·elasticsearch
Cathy Bryant2 小时前
微分几何:曲面与坐标系
笔记·矩阵·高等数学·物理学·微分几何
中屹指纹浏览器2 小时前
2026浏览器指纹追踪的合规边界与隐私优先的反检测技术落地框架
经验分享·笔记
阿Y加油吧2 小时前
算法实战笔记:LeetCode 31 下一个排列 & 287 寻找重复数
笔记·算法·leetcode
就叫飞六吧2 小时前
基于googleshell下载文件/资源
笔记
lwf0061643 小时前
逻辑回归学习笔记-数学直接解回归方程
笔记·学习·逻辑回归
xuhaoyu_cpp_java4 小时前
MyBatis学习(二)
java·经验分享·笔记·学习·mybatis
我登哥MVP5 小时前
【SpringMVC笔记】 - 11 - SpringMVC 执行流程
java·spring boot·笔记·spring·tomcat·intellij-idea
道长爱睡懒觉5 小时前
蓝牙,导航,仪表,TBOX,OTA
笔记
Cathy Bryant5 小时前
微分几何:度规和高斯曲率
笔记·高等数学·物理·微分几何