Udostępnij przez

struktura DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC (directml.h)

Wykonuje funkcję aktywacji softmax na InputTensor, umieszczając wynik w odpowiednim elemecie OutputTensor.

For 1-D InputTensor:
// Let x[i] to be the current element in the InputTensor, and j be the total number of elements in the InputTensor
f(x[i]) = exp(x[i]) / sum(exp(x[0]), ..., exp(x[j-1]))

Gdzie exp(x) jest naturalną funkcją wykładniczą.

Ważne

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

Składnia

struct DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    UINT AxisCount;
    _Field_size_(AxisCount) const UINT* Axes;
};

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.

AxisCount

Typ: UINT

Liczba osi do obliczenia sumy redukcji. To pole określa rozmiar tablicy Osi.

Axes

Typ: _Field_size_(AxisCount) const UINT*

Osie, wzdłuż których można zmniejszyć sumę. Wartości muszą znajdować się w zakresie [0, InputTensor.DimensionCount - 1].

Uwagi

Ten operator jest odpowiednikiem DML_ACTIVATION_SOFTMAX_OPERATOR_DESC , gdy AxisCount == 1 i osie == {DimensionCount - 1}.

Dostępność

Ten operator został wprowadzony w DML_FEATURE_LEVEL_5_1.

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

Zobacz także

  • Last updated on