Compartir a través de

estructura DML_ACTIVATION_PARAMETERIZED_RELU_OPERATOR_DESC (directml.h)

  • Artículo

Realiza una función de activación de unidad lineal rectificada con parámetros (ReLU) en cada elemento de InputTensor, colocando el resultado en el elemento correspondiente de OutputTensor.

f(x, slope) = x,         if x >= 0
              slope * x, otherwise

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

Sintaxis

struct DML_ACTIVATION_PARAMETERIZED_RELU_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *SlopeTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

Tensor de entrada del que se va a leer.

SlopeTensor

Tipo: const DML_TENSOR_DESC*

Tensor que contiene la pendiente para cada valor correspondiente de la entrada.

OutputTensor

Tipo: const DML_TENSOR_DESC*

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

Requisitos

   
Encabezado directml.h

Consulte también

DML_ACTIVATION_RELU_OPERATOR_DESC

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_1_0.

Restricciones tensor

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

Compatibilidad con Tensor

DML_FEATURE_LEVEL_5_1 y versiones posteriores

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

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
SlopeTensor 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
SlopeTensor 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
SlopeTensor Entrada 4 FLOAT32, FLOAT16
OutputTensor Resultados 4 FLOAT32, FLOAT16