estructura DML_ACTIVATION_HARDMAX_OPERATOR_DESC (directml.h)

Realiza una función hardmax en cada elemento de InputTensor, colocando el resultado en el elemento correspondiente de OutputTensor.

El operador calcula el valor hardmax (1 para la primera aparición del valor más grande en la capa y 0 para todos los demás valores) de cada fila de la entrada especificada.

Sintaxis

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

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensor desde el que se va a leer para la entrada. 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 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.

Comentarios

El operador calcula el valor hardmax (1 para el primer valor máximo y 0 para todos los demás) para cada capa del lote de la entrada especificada. La entrada es un tensor 2D (Tensor) de tamaño (batch_size x input_feature_dimensions). El tensor de salida tiene la misma forma y contiene los valores hardmax de la entrada correspondiente.

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_ARGMAX_OPERATOR_DESC