Поделиться через


структура DML_ACTIVATION_GELU_OPERATOR_DESC (directml.h)

Выполняет функцию активации линейной единицы ошибки gaussian (GELU) для каждого элемента в InputTensor, помещая результат в соответствующий элемент OutputTensor.

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

Где DML_ELEMENT_WISE_ERF_OPERATOR_DESC erf(x).

Это важно

Этот API доступен в составе автономного распространяемого пакета DirectML (см. статью Microsoft.AI.DirectML версии 1.9 и более поздних версий. Также см. журнал версий DirectML.

Синтаксис

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

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

Входной тензор для чтения из.

OutputTensor

Тип: const DML_TENSOR_DESC*

Выходной тензор для записи результатов.

Доступность

Этот оператор появился в DML_FEATURE_LEVEL_5_1.

Ограничения Tensor

InputTensor и OutputTensor должны иметь одинаковые значения DataType, DimensionCount и Sizes.

Поддержка Tensor

Тензор Добрый Поддерживаемые счетчики измерений Поддерживаемые типы данных
Входной тензор Ввод От 1 до 8 FLOAT32, FLOAT16
OutputTensor (Тензор выхода) Выходные данные От 1 до 8 FLOAT32, FLOAT16

Требования

   
Заголовок directml.h