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

相关推荐
Brilliantwxx16 分钟前
【算法从零到千】【1-7】 双指针算法
开发语言·c++·笔记·算法·leetcode·推荐算法
chushiyunen24 分钟前
elasticsearch内置接口笔记
大数据·笔记·elasticsearch
问心无愧051340 分钟前
ctf show web入门68,69
android·前端·笔记
江华森42 分钟前
Git + Maven Java 项目部署实战全指南
运维·笔记·git·学习·maven
江屿风1 小时前
C++OJ题经验总结(竞赛)4
开发语言·c++·笔记·算法·dp·双指针
小陈phd1 小时前
多模态大模型学习笔记(四十三)—— 视觉定位(Visual Grounding):语言描述在图像中的精准锚定
笔记·学习·目标跟踪
searchforAI1 小时前
怎么把视频里的PPT提取出来?视频转图文笔记完整方案
人工智能·笔记·gpt·ai·音视频·语音识别·ppt
東雪木12 小时前
多线程与并发编程 专属复习笔记
java·开发语言·笔记·java面试
Oll Correct12 小时前
实验二十九:TCP的运输连接管理
网络·笔记
飞翔中文网13 小时前
Java学习笔记之抽象类与接口(设计思想)
java·笔记·学习