Share via

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

  • Статья

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

f(input, exponent) = pow(input, exponent)

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

Если входной и экспонентный тензор имеют целочисленный тип данных, этот оператор гарантирует точные результаты.

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

Синтаксис

struct DML_ELEMENT_WISE_POW_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *ExponentTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_SCALE_BIAS  *ScaleBias;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

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

ExponentTensor

Тип: const DML_TENSOR_DESC*

Тензор, содержащий значения экспоненты.

OutputTensor

Тип: const DML_TENSOR_DESC*

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

ScaleBias

Тип: _Maybenull_ const DML_SCALE_BIAS*

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

Комментарии

До DML_FEATURE_LEVEL_3_0, ExponentTensorдолжен соответствовать типу InputTensor.

См . DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC для оператора POW, который принимает константу FLOAT для экспоненты.

Доступность

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

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

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

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

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_1_0 и выше

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

Требования

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

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

DML_ELEMENT_WISE_CONSTANT_POW_OPERATOR_DESC