estrutura DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC (directml.h)

Executa uma função de ativação softmax no InputTensor, colocando o resultado no elemento correspondente de 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]))

Onde exp(x) é a função de exponencialização natural.

Sintaxe

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

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de entrada do qual ler.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de saída para o qual gravar os resultados.

AxisCount

Tipo: UINT

O número de eixos a serem calculados. Esse campo determina o tamanho da matriz De eixos .

Axes

Tipo: _Field_size_(AxisCount) const UINT*

Os eixos ao longo dos quais reduzir a soma. Os valores devem estar no intervalo [0, InputTensor.DimensionCount - 1].

Observações

Esse operador é equivalente a DML_ACTIVATION_SOFTMAX_OPERATOR_DESC quando AxisCount == 1 e Eixos == {DimensionCount - 1}.

Disponibilidade

Esse operador foi introduzido no DML_FEATURE_LEVEL_5_1.

Restrições do Tensor

InputTensor e OutputTensor devem ter os mesmos DataType, DimensionCount e Sizes.

Suporte ao Tensor

Requisitos

Consulte também

• DML_ACTIVATION_LOG_SOFTMAX1_OPERATOR_DESC