How to assign NaN to tensor element?

user_15217321 0
Viewed 5251

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.

pytorch python
1 Answers

The following code will set the desired value to nan:

import torch
x = torch.tensor([1, 2, 3]).float()
x[x == 2] = float('nan')
user_66761390
Related