Created by: chunfeng0301

v1 mask1 = (rho < -l_t * grad).detach() v1[mask1] = (l_t * grad2_x)[mask1]

有能实现如上的简洁方法么，不能索引赋值操作