Partilhar via


DML_ACTIVATION_GELU_OPERATOR_DESC estrutura (directml.h)

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

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 do DirectML (consulte Microsoft.AI.DirectML versão 1.9 e posterior. Consulte 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*

O tensor de entrada para ler.

OutputTensor

Tipo: const DML_TENSOR_DESC*

O tensor de saída para gravar os resultados.

Disponibilidade

Este operador foi introduzido em DML_FEATURE_LEVEL_5_1.

Restrições de tensores

InputTensor e OutputTensor devem ter o mesmo DataType, DimensionCount e Sizes.

Suporte tensor

Tensor Amável Contagens de dimensões suportadas Tipos de dados suportados
InputTensor Entrada 1 a 8 FLOAT32, FLOAT16
OutputTensor Resultado 1 a 8 FLOAT32, FLOAT16

Requerimentos

   
Cabeçalho DirectML.H