estructura DML_ACTIVATION_SOFTMAX_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] 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_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	Clase	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	Clase	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	Clase	Recuentos de dimensiones admitidos	Tipos de datos admitidos
InputTensor	Entrada	4	FLOAT32, FLOAT16
OutputTensor	Resultados	4	FLOAT32, FLOAT16

Requisitos

Requisito	Valor
Header	directml.h

Consulte también

estructura de DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC

estructura de DML_ARGMAX_OPERATOR_DESC