Удаление строк в факеле Тензор - PullRequest
Новая
       63

Удаление строк в факеле Тензор

0 голосов
/ 23 октября 2019

У меня есть тензор горелки следующим образом - 

a = tensor(
[[0.2215, 0.5859, 0.4782, 0.7411],
[0.3078, 0.3854, 0.3981, 0.5200],
[0.1363, 0.4060, 0.2030, 0.4940],
[0.1640, 0.6025, 0.2267, 0.7036],
[0.2445, 0.3032, 0.3300, 0.4253]],  dtype=torch.float64)

Если первое значение в каждой строке меньше 0,2, то всю строку необходимо удалить. Таким образом, мне нужен вывод типа - 

tensor(
[[0.2215, 0.5859, 0.4782, 0.7411],
[0.3078, 0.3854, 0.3981, 0.5200],
[0.2445, 0.3032, 0.3300, 0.4253]],  dtype=torch.float64)

Я попытался перебрать тензор и добавить действительное значение в новый пустой тензор, но безуспешно. Есть ли способ получить результаты эффективно?

1 Ответ

1 голос
/ 23 октября 2019

Код 

a = torch.Tensor(
    [[0.2215, 0.5859, 0.4782, 0.7411],
    [0.3078, 0.3854, 0.3981, 0.5200],
    [0.1363, 0.4060, 0.2030, 0.4940],
    [0.1640, 0.6025, 0.2267, 0.7036],
    [0.2445, 0.3032, 0.3300, 0.4253]])

y = a[a[:, 0] > 0.2]
print(y)

Выход 

tensor([[0.2215, 0.5859, 0.4782, 0.7411],
        [0.3078, 0.3854, 0.3981, 0.5200],
        [0.2445, 0.3032, 0.3300, 0.4253]])
...