Udostępnij przez


struktura DML_ACTIVATION_SWISH_OPERATOR_DESC (directml.h)

Wykonuje funkcję aktywacji swish dla każdego elementu w InputTensor, umieszczając wynik w odpowiednim elemecie OutputTensor.

f(x) = x * Sigmoid(SigmoidInputScale * x)

Ten operator obsługuje wykonywanie w miejscu, co oznacza, że tensor wyjściowy jest dozwolony do aliasu InputTensor podczas tworzenia powiązania.

Ważne

Ten interfejs API jest dostępny w ramach autonomicznego pakietu redystrybucyjnego DirectML (zobacz Microsoft.AI.DirectML w wersji 1.13 lub nowszej. Zobacz również historię wersji języka DirectML.

Składnia

struct DML_ACTIVATION_SWISH_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    FLOAT SigmoidInputScale;
};

Członkowie

InputTensor

Typ: const DML_TENSOR_DESC*

Tensor wejściowy do odczytu.

OutputTensor

Typ: const DML_TENSOR_DESC*

Tensor wyjściowy do zapisania wyników.

SigmoidInputScale

Typ: FLOAT

Współczynnik stały. Typową wartością domyślną dla tej wartości jest 1.0.

Dostępność

Ten operator został wprowadzony w DML_FEATURE_LEVEL_6_2.

Ograniczenia dotyczące tensorów

Klasy InputTensor i OutputTensor muszą mieć te same wartości DataType, DimensionCount i Size.

Obsługa biblioteki Tensor

Tensor Rodzaj Obsługiwane liczby wymiarów Obsługiwane typy danych
Tensor wejściowy Dane wejściowe Od 1 do 8 FLOAT32, FLOAT16
Tensor wyjściowy Wynik Od 1 do 8 FLOAT32, FLOAT16

Wymagania

   
Nagłówek directml.h