Po Částech Aktivační Funkce

0

Otázka

Snažím se psát po částech aktivační funkce, jejíž sklon mezi -6 a 0 je 0,1 a ostatní místa jsou jedním. A vstupní(X), size (B, C, H, W). Takže jsem dospěl k závěru, že nejlepší způsob je jednoduchý řádek kódu:

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

Ale jsem čelit tuto chybu:

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

Existuje nějaké řešení pro řešení této chyby?

1

Nejlepší odpověď

1

Nejjednodušší verze toho, co budete potřebovat je:

import torch

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

V jiných jazycích

Tato stránka je v jiných jazycích

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