机器学习|pytorch 中 expand函数和 repeat函数的区别

pytorch 中 expand 和 repeat 的区别
可以参考https://zhuanlan.zhihu.com/p/58109107

再加以补充归纳:
expand 是修改了视图(view),而没有新建内存储存新的值,采用了广播(broadcast)机制。
repeat是按照维度复试了数据,新建了内存去储存新的值。

expand 中 -1代表不扩展维度
repeat中 1 代表不扩展维度
【机器学习|pytorch 中 expand函数和 repeat函数的区别】

    推荐阅读