What is the correct gradient at zero for `pointwise.unit_step`?
I just noticed that the gradient of the pointwise function unit_step
is set to be always zero.
Is that correct, or should we return np.nan
if the input is exactly zero, like for examples sign
does?