Kawałkami Активационная Funkcja

0

Pytanie

Próbuję napisać кусочную funkcję aktywacji, której nachylenie między -6 0 wynosi 0,1, a pozostałe miejsca są równe jeden. I rozmiar danych wejściowych(X) jest równy (B, C, H, W). Dlatego doszedłem do wniosku, że najlepszym sposobem na to prosty kod polecenia:

 x[-6<x and x<0] = x[-6<x and x<0] * 0.1

Ale spotykam się z tym błędem:

RuntimeError: bool value of Tensor with more than one value is ambiguous

Czy istnieje jakieś rozwiązanie aby rozwiązać ten błąd?

1

Najlepsza odpowiedź

1

Najprostsza wersja tego, co potrzebujesz, to:

import torch

def custom_activ(input):
    return torch.where((input>-6) & (input<0.) , 0.1*input, input)
2021-11-23 11:47:17

W innych językach

Ta strona jest w innych językach

Русский
..................................................................................................................
Italiano
..................................................................................................................
Română
..................................................................................................................
한국어
..................................................................................................................
हिन्दी
..................................................................................................................
Français
..................................................................................................................
Türk
..................................................................................................................
Česk
..................................................................................................................
Português
..................................................................................................................
ไทย
..................................................................................................................
中文
..................................................................................................................
Español
..................................................................................................................
Slovenský
..................................................................................................................