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

Статья
02/29/2024

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

f(x) = pow(x, Exponent)

Отрицательные базы поддерживаются для целочисленных экспонент, в противном случае этот оператор возвращает Значение NaN.

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

Синтаксис

struct DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_SCALE_BIAS  *ScaleBias;
  FLOAT                 Exponent;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

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

OutputTensor

Тип: const DML_TENSOR_DESC*

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

ScaleBias

Тип: _Maybenull_ const DML_SCALE_BIAS*

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

Exponent

Тип: FLOAT

Экспонента, к которому будут вызываться все входные данные.

Требования

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

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

DML_ELEMENT_WISE_POW_OPERATOR_DESC

Доступность

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

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

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

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

DML_FEATURE_LEVEL_3_0 и выше

Тензор	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

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