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

Статья
11/12/2024

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

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

Синтаксис

struct DML_ELEMENT_WISE_ROUND_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_ROUNDING_MODE     RoundingMode;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

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

OutputTensor

Тип: const DML_TENSOR_DESC*

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

RoundingMode

Тип: DML_ROUNDING_MODE

DML_ROUNDING_MODE определить направление для округления.

Если DML_ROUNDING_MODE_HALVES_TO_NEAREST_EVEN: значения округляются до ближайшего целого числа, а значения по полпути (например, 0,5) округляются до ближайшего даже целого числа.
Если DML_ROUNDING_MODE_TOWARD_ZERO: значения округляются до нуля. Это эффективно усечение дробной части.
Если DML_ROUNDING_MODE_TOWARD_INFINITY: значения округляются до ближайшего целого числа, при этом пополам (например, 0,5) округляется от нуля (в сторону положительного или отрицательного бесконечности в зависимости от знака значения).

Наличие

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

Ограничения Tensor

InputTensor и outputTensor outputTensor должны иметь одинаковые DataType, DimensionCountи размеры.

Поддержка Tensor

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_2_1 и выше

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

Требования

Требование	Ценность
минимальные поддерживаемые клиентские	Сборка Windows 10 20348
минимальный поддерживаемый сервер	Сборка Windows 10 20348
заголовка	directml.h

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