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

Статья
08/23/2023

Выполняет функцию нормализации Lp вдоль указанной оси входного тензора.

Синтаксис

struct DML_LP_NORMALIZATION_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  UINT                  Axis;
  FLOAT                 Epsilon;
  UINT                  P;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

Тензор, содержащий входные данные.

OutputTensor

Тип: const DML_TENSOR_DESC*

Тензор для записи результатов. Размеры этого тензора должны соответствовать значению InputTensor.

Axis

Тип: UINT

Ось, к которой применяется нормализация.

Epsilon

Тип: FLOAT

Значение epsilon, используемое для предотвращения деления на ноль. По умолчанию рекомендуется использовать значение 0,00001.

P

Тип: UINT

Порядок нормализации (1 или 2).

Доступность

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

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

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

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

DML_FEATURE_LEVEL_3_1 и выше

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

DML_FEATURE_LEVEL_1_0 и выше

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

Требования


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

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