Parçalı Aktivasyon Fonksiyonu

0

Soru

-6 İle 0 arasındaki eğimi 0.1 olan ve diğer yerler bir olan parçalı bir aktivasyon fonksiyonu yazmaya çalışıyorum. Ve giriş (X) boyutu (B, C, H, W) ' dir. Bu yüzden en iyi yolun basit satır kodu olduğu sonucuna vardım:

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

Ama bu hatayla yüzleşiyorum:

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

Bu hatayı çözmek için herhangi bir çözüm var mı?

1

En iyi cevabı

1

İhtiyacınız olan şeyin en basit versiyonu:

import torch

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

Diğer dillerde

Bu sayfa diğer dillerde

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