Par Morceaux De L'Activation De La Fonction

0

La question

Je suis en train d'écrire un par morceaux de fonction d'activation dont la pente entre -6 et 0 0,1 et les autres endroits sont un. Et l'entrée(X) taille de l'est (B, C, H, W). J'en ai donc conclu que le meilleur moyen est de la simple ligne de code:

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

Mais je fais face à cette erreur:

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

Est-il une solution pour résoudre cette erreur?

1

La meilleure réponse

1

La version la plus simple de ce que vous avez besoin est:

import torch

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

Dans d'autres langues

Cette page est dans d'autres langues

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