структура DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC (directml.h)
Выполняет функцию активации с пороговым значением reLU для каждого элемента InputTensor, помещая результат в соответствующий элемент OutputTensor.
f(x) = x, if x > Alpha
0, otherwise
Этот оператор поддерживает выполнение на месте, что означает, что выходному тензору разрешено использовать псевдоним InputTensor во время привязки .
Синтаксис
struct DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC {
const DML_TENSOR_DESC *InputTensor;
const DML_TENSOR_DESC *OutputTensor;
FLOAT Alpha;
};
Члены
InputTensor
Тип: const DML_TENSOR_DESC*
Входной тензор для чтения.
OutputTensor
Тип: const DML_TENSOR_DESC*
Выходной тензор для записи результатов.
Alpha
Тип: FLOAT
Пороговое значение для функции. Обычно для этого значения используется значение 1,0.
Доступность
Этот оператор появился в 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 |