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

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

  • Статья

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

f(x, slope) = x,         if x >= 0
              slope * x, otherwise

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

Синтаксис

struct DML_ACTIVATION_PARAMETERIZED_RELU_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *SlopeTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

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

SlopeTensor

Тип: const DML_TENSOR_DESC*

Тензор, содержащий наклон для каждого соответствующего значения входных данных.

OutputTensor

Тип: const DML_TENSOR_DESC*

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

Требования

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

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

DML_ACTIVATION_RELU_OPERATOR_DESC

Доступность

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

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

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

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

DML_FEATURE_LEVEL_5_1 и выше

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

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_2_0 и выше

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

DML_FEATURE_LEVEL_1_0 и выше

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