12/01/2020

torch.nan_to_num not found error

torch.nan_to_num

This is for 1.8 version.


so use it instead of this.

temp[torch.isnan(temp)] = 0

temp is tensor



No comments:

Post a Comment