I want to assign NaN to a tensor element.
import torch
x = torch.tensor([1, 2, 3])
x[x == 2] = None
I have the error:
TypeError: can't assign a NoneType to a torch.LongTensor
I need it to make sure that some later sophisticated calculations are not made for certain values of x
.
The following code will set the desired value to nan
:
import torch
x = torch.tensor([1, 2, 3]).float()
x[x == 2] = float('nan')