estructura DML_ACTIVATION_SOFTMAX1_OPERATOR_DESC (directml.h)

Realiza una función de activación softmax en InputTensor, colocando el resultado en el elemento correspondiente 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]))

Donde exp(x) es la función de exponente natural.

Sintaxis

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

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensor de entrada desde el que se va a leer.

OutputTensor

Tipo: const DML_TENSOR_DESC*

Tensor de salida en el que se van a escribir los resultados.

AxisCount

Tipo: UINT

Número de ejes que se van a calcular para reducir la suma. Este campo determina el tamaño de la matriz de ejes .

Axes

Tipo: _Field_size_(AxisCount) const UINT*

Ejes a lo largo del cual se reduce la suma. Los valores deben estar en el intervalo [0, InputTensor.DimensionCount - 1].

Observaciones

Este operador es equivalente a DML_ACTIVATION_SOFTMAX_OPERATOR_DESC cuando AxisCount == 1 y == {DimensionCount - 1} ejes.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_5_1.

Restricciones tensor

InputTensor y OutputTensor deben tener el mismo DataType, DimensionCount y Sizes.

Compatibilidad con Tensor

Requisitos

Consulte también

• DML_ACTIVATION_LOG_SOFTMAX1_OPERATOR_DESC