Filter out np.nan values from pytorch 1d tensor

问题

I have a 1d tensor looking kinda like this:

import numpy as np
import torch

my_list = [0, 1, 2, np.nan, np.nan, 4]
tensor = torch.Tensor(my_list)

How do i filter out the nan-values, so it becomes a tensor of size 4?

回答1:

You can use torch.isnan

my_list = [0, 1, 2, np.nan, np.nan, 4]
tensor = torch.Tensor(my_list)

tensor[~torch.isnan(tensor)]
tensor([0., 1., 2., 4.])

来源：https://stackoverflow.com/questions/61503138/filter-out-np-nan-values-from-pytorch-1d-tensor