структура DML_ACTIVATION_PARAMETRIC_SOFTPLUS_OPERATOR_DESC (directml.h)
Выполняет параметрическую функцию активации softplus для каждого элемента в InputTensor, помещая результат в соответствующий элемент OutputTensor.
f(x) = Alpha * ln(1 + exp(Beta * x))
Где exp(x) — это функция естественной экспонентации, а ln(x) — естественный логарифм.
Этот оператор поддерживает выполнение на месте, что означает, что выходному тензору разрешено использовать псевдоним InputTensor во время привязки .
Синтаксис
struct DML_ACTIVATION_PARAMETRIC_SOFTPLUS_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
FLOAT Alpha;
FLOAT Beta;
};
Члены
InputTensor
Тип: const DML_TENSOR_DESC*
Входной тензор для чтения.
OutputTensor
Тип: const DML_TENSOR_DESC*
Выходной тензор для записи результатов.
Alpha
Тип: FLOAT
Коэффициент альфа.
Beta
Тип: FLOAT
Бета-коэффициент.
Доступность
Этот оператор появился в DML_FEATURE_LEVEL_1_0
.
Ограничения тензоров
InputTensor и OutputTensor должны иметь одинаковые значения DataType, DimensionCount и Size.
Поддержка тензоров
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 |