Porțiuni De Activare Funcție

0

Problema

Încerc să scriu o porțiuni de activare funcție a cărei pantă între -6 și 0 este de 0,1 și în alte locuri sunt una. Și de intrare(X) dimensiunea este de (B, C, H, W). Deci am ajuns la concluzia că cel mai bun mod este simpla linie de cod:

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

Dar mă confrunt cu această eroare:

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

Există vreo soluție pentru a rezolva aceasta eroare?

1

Cel mai bun răspuns

1

Cea mai simplă versiune a ceea ce ai nevoie este:

import torch

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

În alte limbi

Această pagină este în alte limbi

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