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

Статья
08/23/2023

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

f(x) = Gamma * x,                        if x > 0
       Gamma * (Alpha * exp(x) - Alpha), otherwise

Где exp(x) — это функция естественной экспоненциации.

Этот оператор поддерживает выполнение на месте, что означает, что выходной тензор допускается для псевдонима InputTensor во время привязки .

Синтаксис

struct DML_ACTIVATION_SCALED_ELU_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Alpha;
  FLOAT                 Gamma;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

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

OutputTensor

Тип: const DML_TENSOR_DESC*

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

Alpha

Тип: FLOAT

Значение alpha. Обычно для этого значения используется значение 1,6732.

Gamma

Тип: FLOAT

Значение гамма. Типичное значение по умолчанию для этого значения — 1,0507.

Доступность

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

Ограничения тензоров

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

Поддержка тензоров

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_2_0 и выше

Тензор	Kind	Поддерживаемые счетчики измерений	Поддерживаемые типы данных
InputTensor	Входные данные	От 4 до 5	FLOAT32, FLOAT16
OutputTensor	Выходные данные	От 4 до 5	FLOAT32, FLOAT16

DML_FEATURE_LEVEL_1_0 и выше

Тензор	Kind	Поддерживаемые счетчики измерений	Поддерживаемые типы данных
InputTensor	Входные данные	4	FLOAT32, FLOAT16
OutputTensor	Выходные данные	4	FLOAT32, FLOAT16

Требования


Верхняя часть	directml.h

См. также раздел

DML_ACTIVATION_ELU_OPERATOR_DESC

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