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

相关推荐
rechol7 分钟前
ARM 架构核心知识笔记(整理与补充版)
arm开发·笔记·架构
paishishaba36 分钟前
JAVA面试复习笔记(待完善)
java·笔记·后端·面试
四谎真好看1 小时前
Java 黑马程序员学习笔记(进阶篇19)
java·笔记·学习·学习笔记
新子y3 小时前
【小白笔记】最大交换 (Maximum Swap)问题
笔记·python
你要飞6 小时前
Hexo + Butterfly 博客添加 Live2D 看板娘指南
笔记
ajsbxi10 小时前
【Java 基础】核心知识点梳理
java·开发语言·笔记
呱呱巨基10 小时前
vim编辑器
linux·笔记·学习·编辑器·vim
新子y10 小时前
【小白笔记】普通二叉树(General Binary Tree)和二叉搜索树的最近公共祖先(LCA)
开发语言·笔记·python
聪明的笨猪猪10 小时前
Java JVM “调优” 面试清单(含超通俗生活案例与深度理解)
java·经验分享·笔记·面试
爱学习的uu10 小时前
CURSOR最新使用指南及使用思路
人工智能·笔记·python·软件工程