How to set 'nan' in Tensor to 0

[justusschock]

For completness I’ll copy my answer:

@justusschock

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