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

相关推荐
思成不止于此7 小时前
MySQL 查询实战(三):排序与综合练习
数据库·笔记·学习·mysql
深海潜水员7 小时前
OpenGL 学习笔记 第一章:绘制一个窗口
c++·笔记·学习·图形渲染·opengl
摇滚侠8 小时前
ElasticSearch 教程入门到精通,文档创建查询修改删除,笔记10、11、12
笔记·elasticsearch
崇山峻岭之间9 小时前
C++ Prime Plus 学习笔记026
c++·笔记·学习
liteblue11 小时前
DEB包解包与打包笔记
linux·笔记
崇山峻岭之间12 小时前
C++ Prime Plus 学习笔记027
c++·笔记·学习
IMPYLH12 小时前
Lua 的 tonumber 函数
开发语言·笔记·后端·junit·游戏引擎·lua
Xudde.12 小时前
BabyPass靶机渗透
笔记·学习·安全·web安全
风123456789~13 小时前
【健康管理学】第10章 中医养生学 1/2
笔记·健康管理
hit56实验室13 小时前
罗计物流为什么会失败
笔记