Compartir a través de


estructura DML_ACTIVATION_SWISH_OPERATOR_DESC (directml.h)

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

f(x) = x * Sigmoid(SigmoidInputScale * x)

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

Importante

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

Sintaxis

struct DML_ACTIVATION_SWISH_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    FLOAT SigmoidInputScale;
};

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.

SigmoidInputScale

Tipo: FLOAT

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

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_6_2.

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