구분적으로 활성화 기능

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ý
..................................................................................................................