Compartir a través de


Estructura DML_ACTIVATION_HARD_SWISH_OPERATOR_DESC (directml.h)

Realiza la función de activación hard swish en cada elemento de InputTensor y coloca el resultado en el elemento correspondiente de OutputTensor.

f(x) = x * HardSigmoid(x, Alpha, Beta)

Este operador admite la ejecución en contexto, lo que significa que el tensor de salida tiene permitido solapar con InputTensor durante el enlace.

Importante

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

Sintaxis

struct DML_ACTIVATION_HARD_SWISH_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    FLOAT Alpha;
    FLOAT Beta;
};

Miembros

InputTensor

Tipo: const DML_TENSOR_DESC*

El tensor de entrada de donde leer.

OutputTensor

Tipo: const DML_TENSOR_DESC*

El tensor de salida en el que escribir el resultado.

Alpha

Tipo: FLOAT

Coeficiente alfa. Un valor predeterminado típico para este valor es 0,2.

Beta

Tipo: FLOAT

: coeficiente de clientes potenciales.

Disponibilidad

Este operador se introdujo en DML_FEATURE_LEVEL_6_2.

Restricciones de tensor

InputTensor y OutputTensor deben tener los mismos DataType, DimensionCount y Sizes.

Compatibilidad del tensor

Tensor Variante Recuentos de dimensiones admitidos Tipos de datos admitidos
InputTensor Entrada 1 a 8 FLOAT32, FLOAT16
OutputTensor Salida 1 a 8 FLOAT32, FLOAT16

Requisitos

   
Encabezado directml.h