Hi, how could you get rid of this issue? I need to use std, is there any solution to avoid nan in the backward pass?