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

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

  • Статья

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

f(a, b) = (a - b) * (a - b)

Этот оператор поддерживает выполнение на месте, то есть OutputTensor во время привязки разрешено использовать псевдонимЫ ATensor или BTensor .

Синтаксис

struct DML_ELEMENT_WISE_DIFFERENCE_SQUARE_OPERATOR_DESC {
  const DML_TENSOR_DESC *ATensor;
  const DML_TENSOR_DESC *BTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Члены

ATensor

Тип: const DML_TENSOR_DESC*

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

BTensor

Тип: const DML_TENSOR_DESC*

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

OutputTensor

Тип: const DML_TENSOR_DESC*

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

Комментарии

Доступность

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

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

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

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

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

Требования

   
Минимальная версия клиента Сборка Windows 22000
Минимальная версия сервера Сборка Windows 22000
Верхняя часть directml.h