torch.repeat 张量维度复制

最新推荐文章于 2025-03-29 19:33:52 发布

陈城南

最新推荐文章于 2025-03-29 19:33:52 发布

阅读量2w

点赞数 9

分类专栏： Pytorch

本文链接：https://blog.csdn.net/qq_40491305/article/details/115110105

版权

Pytorch 专栏收录该内容

12 篇文章

订阅专栏

本文介绍了PyTorch中的torch.repeat函数，用于在指定维度上复制张量。通过示例展示了如何在batch和channel维度上复制张量，帮助理解该函数的工作原理。torch.repeat接受一个张量和一个重复次数的元组，按照元组中每个元素的值在对应维度进行复制。

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

torch.repeat 用于张量的制定维度复制

import torch
a = torch.rand(2,3)  
# 如果 a 是 feature vector, 则 batch=2(dim=0), channels=3(dim=1)
b = a.repeat(2,1)  
# 在 batch(dim=0) 维度上复制 2 次，在 channel(dim=1) 维度上复制 1 次
# 即 repeat 的参数为其维度的复制次数，第 1 个参数表示对第 0 dim 的重复次数，依次类推

########  结果展示  #########

a = torch.rand(2,3)
a
tensor([[0.4040, 0.7043, 0.0454],
        [0.9065, 0.1127, 0.2905]])
a.repeat(2, 1)
tensor([[0.4040, 0.7043, 0.0454],
        [0.9065, 0.1127, 0.2905],
        [0.4040, 0.7043, 0.0454],
        [0.9065, 0.1127, 0.2905]])