torch.repeat函数介绍

repeat

在 PyTorch 中,repeat() 函数用于复制张量的维度。它会将输入张量按照指定的次数重复,以生成一个新的张量。

这是 repeat() 函数的基本语法:

python 复制代码
repeat(*sizes)
  • sizes: 重复每个维度的次数。如果你只想在某些维度上重复,可以在对应位置填入 1

下面是一个简单的例子,说明了 repeat() 函数的用法:

python 复制代码
import torch

# 创建一个张量
x = torch.tensor([[1, 2],
                  [3, 4]])

# 在每个维度上分别重复 2 次
y = x.repeat(2, 2)

print(y)

输出结果是:

复制代码
tensor([[1, 2, 1, 2],
        [3, 4, 3, 4],
        [1, 2, 1, 2],
        [3, 4, 3, 4]])

在这个例子中,原始张量 x 是一个 2x2 的矩阵。通过 x.repeat(2, 2),我们在每个维度上分别重复了 2 次,得到了一个新的 4x4 的张量 y

python 复制代码
a = torch.tensor(
    [[1,2,3],
    [4,5,6]]
)

print(a)


x = a.repeat(1,1)

print(x)

x = a.repeat(1,2)

print(x)

x = a.repeat(2,1)

print(x)
复制代码
tensor([[1, 2, 3],
        [4, 5, 6]])
tensor([[1, 2, 3],
        [4, 5, 6]])
tensor([[1, 2, 3, 1, 2, 3],
        [4, 5, 6, 4, 5, 6]])
tensor([[1, 2, 3],
        [4, 5, 6],
        [1, 2, 3],
        [4, 5, 6]])
相关推荐
27669582928 分钟前
tiktok 弹幕 逆向分析
java·python·tiktok·tiktok弹幕·tiktok弹幕逆向分析·a-bogus·x-gnarly
cylat14 分钟前
Day59 经典时序预测模型3
人工智能·python·深度学习·神经网络
嘉恩督18 分钟前
视频人脸处理——人脸面部动作提取
python·音视频
WJ.Polar19 分钟前
Python数据容器-集合set
开发语言·python
smppbzyc41 分钟前
2025年亚太杯(中文赛项)数学建模B题【疾病的预测与大数据分析】原创论文讲解(含完整python代码)
python·数学建模·数据分析·数学建模竞赛·亚太杯数学建模·亚太杯
xiaocainiao8811 小时前
Python 实战:构建可扩展的命令行插件引擎
开发语言·python
运器1232 小时前
【一起来学AI大模型】PyTorch DataLoader 实战指南
大数据·人工智能·pytorch·python·深度学习·ai·ai编程
音元系统2 小时前
Copilot 在 VS Code 中的免费替代方案
python·github·copilot
超龄超能程序猿2 小时前
(5)机器学习小白入门 YOLOv:数据需求与图像不足应对策略
人工智能·python·机器学习·numpy·pandas·scipy
cooldream20094 小时前
Python 包管理新时代:深入了解 `uv` 的使用与实践
python·uv·包管理器