Estructura DML_ACTIVATION_SWISH_OPERATOR_DESC (directml.h)

Realiza una función de activación swish en cada elemento de InputTensor y coloca 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 tiene permitido solaparse 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_SWISH_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    FLOAT SigmoidInputScale;
};

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.

SigmoidInputScale

Tipo: FLOAT

El 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 de tensor

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

Compatibilidad del tensor

Tensor	Clase	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