笔记68：Pytorch中repeat函数的用法

唔希迪西

于 2023-12-06 19:55:43 发布

阅读量482

点赞数

分类专栏：深度学习文章标签：笔记

版权

34 篇文章

订阅专栏

本文介绍了PyTorch库中`repeat`函数的作用，它用于沿指定维度复制tensor数据，不同于`expand`，`repeat`复制的是tensor内容。通过例子展示了如何使用`repeat`并打印出其改变后的tensor形状。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

repeat 相当于一个broadcasting的机制

repeat(*sizes)

沿着指定的维度重复tensor。不同与expand()，本函数复制的是tensor中的数据。

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)

torch.Size([128, 5, 512])