Кусочно-Активационная Функция

0

Вопрос

Я пытаюсь написать кусочную функцию активации, наклон которой между -6 и 0 равен 0,1, а остальные места равны одному. И размер входного сигнала(X) равен (B, C, H, W). Поэтому я пришел к выводу, что лучший способ-это простой код строки:

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

Но я сталкиваюсь с этой ошибкой:

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

Есть ли какое-либо решение для устранения этой ошибки?

1

Лучший ответ

1

Самая простая версия того, что вам нужно, это:

import torch

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

На других языках

Эта страница на других языках

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