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

相关推荐
thisbrand10 小时前
李辉《曾国藩日记》笔记:放不下对诗的情节
笔记·曾国藩
摇滚侠13 小时前
Java 零基础全套教程,File 类与 IO 流,笔记 177-178
java·开发语言·笔记
观浩13 小时前
使用Obsidian和IDEA搭建个人知识库
经验分享·笔记·学习方法
玄米乌龙茶12315 小时前
LLM成长笔记(十三):系统设计与项目实战
笔记·unity·游戏引擎
searchforAI16 小时前
AI多模态技术:从语音识别到AI结构化笔记是怎么实现的
人工智能·经验分享·笔记·gpt·whisper·语音识别
凉、介16 小时前
深入理解 ARMv7-A|异常/中断处理
笔记·学习·嵌入式·arm
koo36416 小时前
周报5.24
笔记
玄米乌龙茶12317 小时前
LLM成长笔记(十一):模型部署与工程化
笔记
会编程的土豆17 小时前
结构体标签与数据流向 笔记
笔记
玄米乌龙茶12318 小时前
LLM成长笔记(十):多模态应用开发
人工智能·笔记·语音识别