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

Статья
08/23/2023

Выполняет следующую операцию для каждого элемента InputTensor, помещая результат в соответствующий элемент OutputTensor. Этот оператор зажимает (или ограничивает) каждый элемент во входных данных в пределах закрытого интервала [Min, Max].

f(x) = max(Min, min(x, Max))

Где max(a,b) возвращает большее из двух значений, а min(a,b) возвращает меньшее из двух значений a,b.

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

Синтаксис

struct DML_ELEMENT_WISE_CLIP1_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_SCALE_BIAS  *ScaleBias;
  DML_TENSOR_DATA_TYPE  MinMaxDataType;
  DML_SCALAR_UNION      Min;
  DML_SCALAR_UNION      Max;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

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

OutputTensor

Тип: const DML_TENSOR_DESC*

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

ScaleBias

Тип: _Maybenull_ const DML_SCALE_BIAS*

Необязательный масштаб и смещение, применяемые к входным данным. Если он имеется, функция применяется g(x) = x * scale + bias к каждому входным элементу перед вычислением этого оператора.

MinMaxDataType

Тип: DML_TENSOR_DATA_TYPE

Тип данных элементов Min и Max , которые должны соответствовать OutputTensor.DataType.

Min

Тип: DML_SCALAR_UNION

Минимальное значение, ниже которого оператор заменяет значение min. MinMaxDataType определяет способ интерпретации поля.

Max

Тип: DML_SCALAR_UNION

Максимальное значение, выше которого оператор заменяет значение max. MinMaxDataType определяет способ интерпретации поля.

Доступность

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

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

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

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

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

Требования


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

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