【笔记】pytorch语法 torch.repeat & torch.expand

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