Отфильтровать значения np.nan из 1d тензора pytorch

Question

У меня есть 1d-тензор, который выглядит примерно так:

import numpy as np
import torch

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

Как мне отфильтровать нан-значения, чтобы он стал тензором размера 4?

Answer 1

Вы можете использовать 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.])