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

Статья
02/29/2024

Выполняет функцию масштабирования и смещения на уровне элементов, Output = Scale * Input + Bias. Этот оператор аналогичен использованию DML_ELEMENT_WISE_IDENTITY_OPERATOR_DESC с масштабом и смещением, за исключением того, что DML_VALUE_SCALE_2D_OPERATOR_DESC применяет разные смещения для каждого канала, а не одиночный смещение для всего тензора.

Синтаксис

struct DML_VALUE_SCALE_2D_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Scale;
  UINT                  ChannelCount;
  const FLOAT           *Bias;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

Тензор, содержащий входные данные. Измерения этого тензора должны иметь значение { BatchCount, ChannelCount, Height, Width }.

OutputTensor

Тип: const DML_TENSOR_DESC*

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

Scale

Тип: FLOAT

Масштабируемое значение, применяемое ко всем входным значениям.

ChannelCount

Тип: UINT

Это поле определяет размер массива Bias. Это поле должно иметь значение 1 или 3, а также должно соответствовать размеру измерения Канала входного тензора.

Bias

Тип: const FLOAT*

Массив значений FLOAT , содержащий термин смещения для каждого измерения входного тензора.

Доступность

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

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

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

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

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

Требования

Требование	Значение
Заголовок	directml.h

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