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

相关推荐
Hello_Embed12 分钟前
libmodbus 移植到 STM32H5
笔记·stm32·单片机·嵌入式硬件·嵌入式·ai编程
Biocloudy25 分钟前
循环肿瘤细胞的分离和分型技术
人工智能·经验分享·笔记·其他
Undergoer_TW40 分钟前
【SLAM性能评估笔记】公开的Vo性能评估工具调研与局限性分析
笔记·evo·kitti·vo·性能评估·tum
空太Jun41 分钟前
Git 使用学习笔记
笔记·git·学习
智者知已应修善业1 小时前
【51单片机独立按键控制数码管自增自减】2023-10-5
c++·经验分享·笔记·算法·51单片机
SHARK_pssm1 小时前
【数据结构——复杂度】
c语言·数据结构·经验分享·笔记
中屹指纹浏览器2 小时前
2026指纹浏览器轻量化架构与资源调度技术:实现千级环境高效稳定运行
经验分享·笔记
咸甜适中2 小时前
rust语言学习笔记Trait之 From 和 Into (类型转换)
笔记·学习·rust
叶~小兮2 小时前
K8S优先级、Pod驱逐、HPA扩缩容 学习笔记
笔记·学习·kubernetes
星恒随风2 小时前
四天学完前端基础三件套(CSS篇)
前端·css·笔记·学习