torch:expand和repeate的区别

1. expand -- 不复制数据的广播式扩展

  • 规则 :只能将 size=1 的维度扩展为更大。

  • 内存 :返回视图(与原张量共享数据),高效。

  • 语法tensor.expand(*target_shape)

复制代码
import torch

x = torch.tensor([[1], [2], [3]])   # (3, 1)
y = x.expand(3, 4)                  # (3, 4) 把第二维从1扩展到4
# y: [[1,1,1,1], [2,2,2,2], [3,3,3,3]]

# 支持 -1 表示保持该维度大小
z = x.expand(3, -1)                 # (3, 1) 保持不变

注意:通过将目标形状对应位置设为 1 或使用 unsqueeze 配合

复制代码
# 想要与形状为 (batch, 3, H, W) 的输入广播,但 weight 没有 batch 维度
weight_expanded = weight.unsqueeze(0).expand(8, -1, -1, -1, -1)  
# shape: (8, 64, 3, 5, 5) ,其中 -1 表示保持原大小

2. repeat -- 实际复制数据的重复平铺

  • 规则:将张量在每个维度上重复指定次数(次数是整数)。

  • 内存 :分配新内存,独立于原张量。

  • 语法tensor.repeat(*repeats) -- 重复次数列表。

复制代码
x = torch.tensor([[1, 2], [3, 4]])   # (2, 2)
y = x.repeat(2, 3)                   # (4, 6) 第0维重复2次,第1维重复3次
# y:
# [[1,2,1,2,1,2],
#  [3,4,3,4,3,4],
#  [1,2,1,2,1,2],
#  [3,4,3,4,3,4]]

关键区别速记表

函数 限制 内存 典型场景
expand 只能扩展 size=1 的维度 共享 偏置加法、广播对齐
repeat 可重复任意维度 复制 需要独立副本、平铺图案
  • 只想逻辑上"变大"用 expand(省内存)。

  • 需要真正复制数据用 repeat(独立存储)。

相关推荐
灯澜忆梦3 小时前
GO_并发编程---定时器
开发语言·后端·golang
-银雾鸢尾-3 小时前
C#中的StringBuilder相关方法
开发语言·c#
-银雾鸢尾-4 小时前
C#中结构体与类的区别;抽象类与接口的区别
开发语言·c#
大模型码小白5 小时前
【Python零基础教程】继承、多态与魔法函数:面向对象编程三大核心特性详解
java·大数据·开发语言·人工智能·python·ai编程
麻雀飞吧5 小时前
最新量化学习路径,交易认知和技术实现要并行
人工智能·python
段一凡-华北理工大学7 小时前
向量数据库实战:选型、调优与落地~系列文章12:文本分块策略实战:chunk_size 怎么选?重叠多少?
开发语言·数据库·后端·oracle·rust·工业智能体·高炉智能化
Ljwuhe7 小时前
C++——多态
开发语言·c++
心平气和量大福大7 小时前
C#-WPF-Window主窗体
开发语言·c#·wpf
从零开始的代码生活_9 小时前
C++ 继承详解:访问控制、对象模型、菱形继承与设计取舍
开发语言·c++·后端·学习·算法
云小逸9 小时前
【C++ 第七阶段:模板、泛型编程与工程综合详解】
开发语言·c++