estructura DML_ACTIVATION_CELU_OPERATOR_DESC (directml.h)

Artículo
08/23/2023

Realiza la función de activación de unidad lineal exponencial (CELU) continuamente diferente en cada elemento de InputTensor, colocando el resultado en el elemento correspondiente de OutputTensor.

f(x) = max(0, x) + min(0, Alpha * (exp(x / Alpha) - 1));

Donde:

exp(x) es la función de exponente natural.
max(a,b) devuelve el mayor de los dos valores a,b
min(a,b) devuelve el menor de los dos valores a,b

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_CELU_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Alpha;
};

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.

Alpha

Tipo: FLOAT

Coeficiente alfa. Un valor predeterminado típico para este valor es 1.0.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_3_0.

Restricciones tensor

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

Compatibilidad con Tensor

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

Requisitos


Cliente mínimo compatible	Compilación 20348 de Windows 10
Servidor mínimo compatible	Compilación 20348 de Windows 10
Encabezado	directml.h

Share via