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

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

  • Статья

Вычисляет градиенты обратного распространения для элементного клипа.

f(x, gradient) = if x <= Min then 0
                 if x >= Max then 0
                 else        then gradient

Этот оператор поддерживает выполнение на месте, то есть OutputTensor во время привязки разрешено использовать псевдоним InputTensor .

Синтаксис

struct DML_ELEMENT_WISE_CLIP_GRAD_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *InputGradientTensor;
  const DML_TENSOR_DESC *OutputGradientTensor;
  FLOAT                 Min;
  FLOAT                 Max;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

Тензор входной функции. Обычно это тот же тензор, который был указан в качестве объекта InputTensor для DML_ELEMENT_WISE_CLIP_OPERATOR_DESC на переднем проходе.

InputGradientTensor

Тип: const DML_TENSOR_DESC*

Тензор входящего градиента. Обычно это получается из выходных данных обратного распространения предыдущего слоя. Обычно этот тензор имеет те же размеры, что и выходные данные соответствующего DML_OPERATOR_ELEMENT_WISE_CLIP на переднем проходе.

OutputGradientTensor

Тип: const DML_TENSOR_DESC*

Выходной тензор, содержащий обратные градиенты. Обычно этот тензор имеет те же размеры, что и входные данные соответствующего DML_OPERATOR_ELEMENT_WISE_CLIP на переднем проходе.

Min

Тип: FLOAT

Минимальное значение. Если значение x равно этому значению или ниже, результат градиента равен 0.

Max

Тип: FLOAT

Максимальное значение. Если значение x равно этому значению или выше, результат градиента равен 0.

Комментарии

Доступность

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

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

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

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

DML_FEATURE_LEVEL_5_0 и выше

Тензор Kind Поддерживаемые счетчики измерений Поддерживаемые типы данных
InputTensor Входные данные От 1 до 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
InputGradientTensor Входные данные От 1 до 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8
OutputGradientTensor Выходные данные От 1 до 8 FLOAT32, FLOAT16, INT64, INT32, INT16, INT8, UINT64, UINT32, UINT16, UINT8

DML_FEATURE_LEVEL_3_1 и выше

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

Требования

   
Минимальная версия клиента Сборка Windows 22000
Минимальная версия сервера Сборка Windows 22000
Верхняя часть directml.h