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

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

  • Статья

Вычисляет арктангенс из двух аргументов для каждого элемента ATensor и BTensor, где ATensor — ось Y , а BTensor — ось X, помещая результат в соответствующий элемент OutputTensor. Этот оператор не определен для источника (т. е. если ATensor и BTensor имеют значение 0 для соответствующих элементов).

GRU_Forward

f(y, x) = atan2(y, x)

Этот оператор поддерживает выполнение на месте, что означает, что выходной тензор может быть псевдонимом ATensor или BTensor во время привязки .

Синтаксис

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

Члены

ATensor

Тип: const DML_TENSOR_DESC*

Входной тензор для считывания значений оси Y.

BTensor

Тип: const DML_TENSOR_DESC*

Входной тензор для считывания значений оси X.

OutputTensor

Тип: const DML_TENSOR_DESC*

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

Комментарии

Доступность

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

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

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

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

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

Требования

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