ReLU turns nan into zeros

[justusschock]

since NaN != NaN you could do my_tensor[my_tensor!=my_tensor] = 0