Compartir a través de

estructura DML_ACTIVATION_SHRINK_OPERATOR_DESC (directml.h)

Realiza la función de activación de reducción en cada elemento de InputTensor, colocando el resultado en el elemento correspondiente de OutputTensor.

f(x) = x - Bias, if x > Threshold
       x + Bias, if x < -Threshold
       0,        otherwise

Este operador admite la ejecución local, lo que significa que el tensor de salida puede alias InputTensor durante el enlace.

Sintaxis

struct DML_ACTIVATION_SHRINK_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Bias;
  FLOAT                 Threshold;
};

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.

Bias

Tipo: FLOAT

Valor del sesgo. Un valor predeterminado típico para este valor es 0,0.

Threshold

Tipo: FLOAT

Valor del umbral. Un valor predeterminado típico para este valor es 0,5.

Comentarios

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_2_0.

Restricciones tensor

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

Compatibilidad con Tensor

DML_FEATURE_LEVEL_5_1 y versiones posteriores

Tensor Clase Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada De 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8
OutputTensor Resultados De 1 a 8 FLOAT32, FLOAT16, INT32, INT16, INT8, UINT32, UINT16, UINT8

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

Requisitos

Requisito Value
Cliente mínimo compatible Windows 10, versión 2004 (10.0; Compilación 19041)
Servidor mínimo compatible Windows Server, versión 2004 (10.0; Compilación 19041)
Encabezado directml.h
  • Last updated on