pytorch 中 expand 和 repeat 的区别
可以参考https://zhuanlan.zhihu.com/p/58109107
再加以补充归纳:
expand 是修改了视图(view),而没有新建内存储存新的值,采用了广播(broadcast)机制。
repeat是按照维度复试了数据,新建了内存去储存新的值。
expand 中 -1代表不扩展维度
repeat中 1 代表不扩展维度
再加以补充归纳:
expand 是修改了视图(view),而没有新建内存储存新的值,采用了广播(broadcast)机制。
repeat是按照维度复试了数据,新建了内存去储存新的值。
expand 中 -1代表不扩展维度
repeat中 1 代表不扩展维度