显存爆炸的问题 a[a>=b]=1
收藏
a[a>=b]=x 会导致显存一直积累, 可能是一直产生a 的拷贝
使用 paddle.where(a>=b, x1, x2) 来替代
0
收藏
请登录后评论
厉害
厉害
厉害