DML_ACTIVATION_SWISH_OPERATOR_DESC estrutura (directml.h)

Executa uma função de ativação de swish em cada elemento em InputTensor, colocando o resultado no elemento correspondente de OutputTensor.

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

Esse operador dá suporte à execução in-loco, o que significa que o tensor de saída tem permissão para alias InputTensor durante a associação.

Sintaxe

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

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de entrada do qual ler.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de saída para o qual gravar os resultados.

SigmoidInputScale

Tipo: FLOAT

O coeficiente constante. Um padrão típico para esse valor é 1,0.

Disponibilidade

Esse operador foi introduzido no DML_FEATURE_LEVEL_6_2.

Restrições do Tensor

InputTensor e OutputTensor devem ter os mesmos DataType, DimensionCount e Sizes.

Suporte ao Tensor

Requisitos