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

Статья
11/12/2024

Проверяет каждый элемент InputTensor для IEEE-754 -inf, inf или обоих, в зависимости от заданного InfinityMode, и помещает результат (1 для true, 0 для false) в соответствующий элемент OutputTensor.

f(x) = isinf(x) && (
       (x > 0 && InfinityMode == DML_IS_INFINITY_MODE_POSITIVE) ||
       (x < 0 && InfinityMode == DML_IS_INFINITY_MODE_NEGATIVE) ||
                 InfinityMode == DML_IS_INFINITY_MODE_EITHER)

Синтаксис

struct DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_IS_INFINITY_MODE  InfinityMode;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

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

OutputTensor

Тип: const DML_TENSOR_DESC*

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

InfinityMode

Тип: DML_IS_INFINITY_MODE

DML_IS_INFINITY_MODE определить знак бесконечности для проверки.

Если DML_IS_INFINITY_MODE_EITHER, то 1 будет возвращен, если элемент -inf или inf, в противном случае — значение 0.
Если DML_IS_INFINITY_MODE_POSITIVE, 1 будет возвращено, если элемент имеет значение inf, в противном случае — значение 0.
Если DML_IS_INFINITY_MODE_NEGATIVE', 1 будет возвращено, если элемент имеет значение -inf, в противном случае — 0.

Замечания

Наличие

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

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

InputTensor и OutputTensor должны иметь одинаковые размеров иразмеров.

Поддержка Tensor

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_2_1 и выше

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

Требования

Требование	Ценность
минимальные поддерживаемые клиентские	Windows 10 версии 2004 (10.0; Сборка 19041)
минимальный поддерживаемый сервер	Windows Server версии 2004 (10.0; Сборка 19041)
заголовка	directml.h

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