Piecewise Aktivácie Funkcie

0

Otázka

Snažím sa napísať piecewise aktivácie funkcie, ktorých svahu medzi -6 a 0 je 0,1 a iných miestach, ktoré sú jedným. A vstup(X) veľkosť je (B, C, H, W). Tak som dospela k záveru, že najlepší spôsob je jednoduchý riadok kódu:

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

Ale ja tvár túto chybu:

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

Existuje nejaké riešenie pre riešenie tejto chyby?

1

Najlepšiu odpoveď

1

Najjednoduchšiu verziu, čo potrebujete, 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 iných jazykoch

Táto stránka je v iných jazykoch

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