pytorch 中 expand函数和 repeat函数的区别

pytorch 中 expand 和 repeat 的区别

 

可以参考https://zhuanlan.zhihu.com/p/58109107

 

再加以补充归纳：

expand 是修改了视图（view），而没有新建内存储存新的值，采用了广播（broadcast）机制。

repeat是按照维度复试了数据，新建了内存去储存新的值。

 

expand 中 -1代表不扩展维度

repeat中 1 代表不扩展维度