A Tratti Funzione Di Attivazione

0

Domanda

Sto cercando di scrivere un piecewise funzione di attivazione la cui pendenza tra -6 e 0 0,1 e in altri luoghi sono uno. E l'ingresso(X) dimensioni (B, C, H, W). Così ho concluso che il modo migliore è la semplice riga di codice:

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

Ma mi faccia questo errore:

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

C'è qualche soluzione per risolvere questo errore?

1

Migliore risposta

1

Il più semplice di quello che vi serve è:

import torch

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

In altre lingue

Questa pagina è in altre lingue

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