【笔记】pytorch语法 torch.repeat & torch.expand
发布时间：2023-06-23 17:30
torch.repeat 与 torch.expand 比较
torch.repeat和torch.expand的区别：
与numpy类似函数的比较
torch.repeat
torch.repeat用法类似np.tile，同时numpy中也有一个repeat，用法对应torch中的 torch.repeat_interleave。
这两者的类似都是在参数层面和使用方法上的类似，torch.repeat和torch.repeat_interleave得到的结果与输入都没有关联。
torch.expand
torch.expand的结果与输入有关联，见上图中例子，类似numpy.broadcast_to。同时numpy.broadcast_to在使用方法上类似np.tile。
总结
torch.repeat与torch.expand的差异，类似np.tile与np.broadcast_to的差异。
| torch | numpy | 是否broadcast | 参数类型和使用方法 |
|---|---|---|---|
| torch.repeat | np.tile | 否 | A |
| torch.expand | np.broadcast_to | 是 | A |
| torch.repeat_interleave | np.repeat | 否 | B |