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

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

  • Статья

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

f(a, b) = a + b

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

Синтаксис

struct DML_ELEMENT_WISE_ADD_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_2_0более новая версия этого оператора, DML_ELEMENT_WISE_ADD1_OPERATOR_DESC.

Доступность

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

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

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

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

DML_FEATURE_LEVEL_5_1 и выше

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

DML_FEATURE_LEVEL_4_1 и выше

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

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_2_1 и выше

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

DML_FEATURE_LEVEL_1_0 и выше

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

Требования

   
Верхняя часть directml.h

См. также раздел

DML_ELEMENT_WISE_ADD1_OPERATOR_DESC