Esta Função De Ativação

0

Pergunta

Eu estou tentando escrever um piecewise função de ativação cuja inclinação entre -6 e 0 é igual a 0.1 e os outros lugares que são um. E a entrada(X) tamanho, (B, C, H, W). Então cheguei a conclusão de que a melhor maneira é a simples linha de código:

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

Mas eu encaro esse erro:

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

Há alguma solução para resolver este erro?

1

Melhor resposta

1

A versão mais simples do que você precisa é:

import torch

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

Em outros idiomas

Esta página está em outros idiomas

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