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

相关推荐
AA陈超3 小时前
ASC学习笔记0014:手动添加一个新的属性集
c++·笔记·学习·ue5
Chunyyyen4 小时前
【第二十二周】自然语言处理的学习笔记06
笔记·学习·自然语言处理
谅望者7 小时前
数据分析笔记06:假设检验
笔记·数据挖掘·数据分析
摇滚侠7 小时前
Vue 项目实战《尚医通》,获取当前账户就诊人信息并展示出来,笔记42
前端·javascript·vue.js·笔记·html5
d111111111d8 小时前
STM32通信协议学习--I2C通信(了解)
笔记·stm32·单片机·嵌入式硬件·学习
Rousson10 小时前
硬件学习笔记--89 MCU主频对Uart波特率影响及采样点
笔记·单片机·学习
2301_8018217111 小时前
标签分类调研
笔记
im_AMBER11 小时前
HTTP 02 会话 | 消息 | MIME类型
网络·笔记·网络协议·学习·http
循环过三天11 小时前
7.5、Python-匿名函数lambda
笔记·python·学习
不会写代码的里奇13 小时前
VMware Ubuntu 22.04 NAT模式下配置GitHub SSH完整教程(含踩坑实录+报错_成功信息对照)
linux·经验分享·笔记·git·ubuntu·ssh·github