Estrutura DML_ACTIVATION_GELU_OPERATOR_DESC (directml.h)

Artigo
03/19/2024

Executa uma função de ativação de unidade linear de erro gaussian (GELU) em cada elemento no InputTensor, colocando o resultado no elemento OutputTensor correspondente.

f(x) = 0.5 * x * (1.0 + erf(x / sqrt(2)))

Onde erf(x) é DML_ELEMENT_WISE_ERF_OPERATOR_DESC.

Importante

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

Sintaxe

struct DML_ACTIVATION_GELU_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
};

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.

Disponibilidade

Esse operador foi introduzido em DML_FEATURE_LEVEL_5_1.

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