Compartir a través de


estructura DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC (directml.h)

Realiza una función de activación log-of-softmax natural en cada elemento de InputTensor, colocando el resultado en el elemento correspondiente de OutputTensor.

// Let x_i be the current value in the axis, and j be the total number of elements along that axis.
f(x_i) = ln(exp(x_i) / sum(exp(x_0), ..., exp(x_j)))

Donde exp(x) es la función de exponente natural y ln(x) es el logaritmo natural.

Sintaxis

struct DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensor de entrada del que se va a leer. Este tensor debe tener un rango efectivo no mayor que 2. La clasificación efectiva de un tensor es DimensionCount del tensor, excepto las dimensiones situadas más a la izquierda del tamaño 1. Por ejemplo, un tamaño de tensor de { 1, 1, BatchCount, Width } es válido y es equivalente a un tensor de tamaños { BatchCount, Width }.

OutputTensor

Tipo: const DML_TENSOR_DESC*

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

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_1_0.

Restricciones tensor

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

Compatibilidad con Tensor

DML_FEATURE_LEVEL_3_0 y versiones posteriores

Tensor Kind Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 1 a 8 FLOAT32, FLOAT16
OutputTensor Resultados De 1 a 8 FLOAT32, FLOAT16

DML_FEATURE_LEVEL_2_0 y versiones posteriores

Tensor Kind Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 4 a 5 FLOAT32, FLOAT16
OutputTensor Resultados De 4 a 5 FLOAT32, FLOAT16

DML_FEATURE_LEVEL_1_0 y versiones posteriores

Tensor Kind Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada 4 FLOAT32, FLOAT16
OutputTensor Resultados 4 FLOAT32, FLOAT16

Requisitos

   
Encabezado directml.h

Consulte también

estructura de DML_ACTIVATION_SOFTMAX_OPERATOR_DESC

estructura de DML_ARGMAX_OPERATOR_DESC