Compartir a través de


estructura DML_ACTIVATION_GELU_OPERATOR_DESC (directml.h)

Realiza la función de activación de unidad lineal de error gaussiano (GELU) en cada elemento de InputTensor, colocando el resultado en el elemento correspondiente de OutputTensor.

f(x) = 0.5 * x * (1.0 + erf(x / sqrt(2)))

Donde erf(x) es DML_ELEMENT_WISE_ERF_OPERATOR_DESC.

Importante

Esta API está disponible como parte del paquete redistribuible independiente de DirectML (consulte Microsoft.AI.DirectML versión 1.9 y posteriores). Consulte también historial de versiones de DirectML.

Sintaxis

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

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.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_5_1.

Restricciones tensor

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

Compatibilidad con Tensor

Tensor Variante Recuentos de dimensiones admitidos Supported data types (Tipos de datos admitidos)
InputTensor Entrada De 1 a 8 FLOAT32, FLOAT16
OutputTensor Salida De 1 a 8 FLOAT32, FLOAT16

Requisitos

   
Encabezado directml.h