A Trozos Función De Activación De La

0

Pregunta

Estoy tratando de escribir una trozos de activación de la función cuya pendiente entre -6 y 0 es de 0,1 y los demás lugares son uno. Y la entrada(X) es (B, C, H, W). Así que llegué a la conclusión de que la mejor manera es la simple línea de código:

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

Pero me enfrento a este error:

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

Existe alguna solución para resolver este error?

1

Mejor respuesta

1

La versión más simple de lo que usted necesita es:

import torch

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

En otros idiomas

Esta página está en otros idiomas

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