Estrutura DML_ACTIVATION_SWISH_OPERATOR_DESC (directml.h)

Artigo
03/19/2024

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

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

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.

Importante

Essa API está disponível como parte do pacote redistribuível autônomo DirectML (consulte Microsoft.AI.DirectML versão 1.13 e posterior. Confira também o histórico de versões do DirectML.

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*

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.

SigmoidInputScale

Tipo: FLOAT

O coeficiente constante. Um padrão típico para esse valor é 1.0.

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

Tensor	Tipo	Contagens de dimensões compatíveis	Tipos de dados com suporte
InputTensor	Entrada	1 a 8	FLOAT32, FLOAT16
OutputTensor	Saída	1 a 8	FLOAT32, FLOAT16

Requisitos


Cabeçalho	directml.h

Compartilhar via