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

Статья
02/26/2024

Для каждого элемента входного тензора возвращает значение 1, если входные данные являются NaN (согласно определению IEEE-754), и 0 в противном случае. Результат помещается в соответствующий элемент выходного тензора.

Синтаксис

struct DML_ELEMENT_WISE_IS_NAN_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

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

OutputTensor

Тип: const DML_TENSOR_DESC*

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

Доступность

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

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

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

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

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_2_0 и выше

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

Требования

Требование	Значение
Минимальная версия клиента	Windows 10, версия 2004 (10.0; Сборка 19041)
Минимальная версия сервера	Windows Server версии 2004 (10.0; Сборка 19041)
Верхняя часть	directml.h

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