Estrutura DML_ACTIVATION_HARD_SWISH_OPERATOR_DESC (directml.h)

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

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

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

Sintaxe

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

Membros

InputTensor

Tipo: const DML_TENSOR_DESC*

A entrada do tensor da qual fazer a leitura.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de saída no qual gravar o resultado.

Alpha

Tipo: FLOAT

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

Beta

Tipo: FLOAT

O coeficiente beta.

Disponibilidade

Esse operador foi introduzido em DML_FEATURE_LEVEL_6_2.

Restrições de tensor

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

Suporte a tensores

Requisitos