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

相关推荐
Strugglingler7 分钟前
Linux Device Drivers-第七章 时间, 延迟及延缓操作
linux·笔记
風清掦14 分钟前
【江科大STM32学习笔记-11】SPI通信协议 - 11.2 软件SPI读写W25Q64
笔记·stm32·单片机·嵌入式硬件·学习
ouliten36 分钟前
cuda编程笔记(40)--Pipelines(流水线)
笔记·cuda
w2018001 小时前
段永平投资问答录pdf完整版
笔记·pdf
我的征途是星辰大海。1 小时前
设计模式(学习笔记)(第一章)
笔记·学习·设计模式
弘毅 失败的 mian1 小时前
STM32 时钟详解
经验分享·笔记·stm32·单片机·嵌入式硬件·嵌入式
qeen871 小时前
【算法笔记】差分与经典例题解析
c语言·c++·笔记·学习·算法·差分
中屹指纹浏览器1 小时前
2026分布式多账号运营下指纹浏览器集群调度方案
经验分享·笔记
摇滚侠1 小时前
Java 零基础全套视频教程,面向对象(进阶),笔记 90-103
java·开发语言·笔记
23471021271 小时前
4.22 学习笔记
软件测试·笔记·python·学习