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

相关推荐
许长安8 分钟前
Redis 跳表实现详解
数据库·c++·经验分享·redis·笔记·缓存
June bug36 分钟前
【雅思学习笔记】Part2话题词汇及表达
笔记·学习
Hehuyi_In1 小时前
postgres-howto 学习笔记
笔记·学习·postgresql·脚本·how to
蜗牛^^O^1 小时前
Agent学习笔记
笔记·学习
问心无愧05132 小时前
ctf show web入门90
前端·笔记
NULL指向我2 小时前
STM32 F103C8T6学习笔记20:SPI驱动W25Qxx
笔记·stm32·学习
脆皮炸鸡7553 小时前
进程通信----命名管道
linux·经验分享·笔记·算法·学习方法
Oll Correct3 小时前
实验二十四:网络地址与端口号转换NAPT
网络·笔记
三品吉他手会点灯3 小时前
C语言学习笔记 - 32.嵌入式C语言学习阶段对初学编程者的建议
c语言·开发语言·笔记·学习
数据皮皮侠AI3 小时前
基于经济学季刊方法测算的中国城市蔓延指数
大数据·人工智能·笔记·数据挖掘·回归