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

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

  • Статья

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

f(a, b) = (a >> b)

Побитовая операция применяется к тензорным данным в собственной кодировке. Таким образом, тензорный тип данных игнорируется, за исключением определения ширины каждого элемента.

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

Синтаксис

struct DML_ELEMENT_WISE_BIT_SHIFT_RIGHT_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_1.

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

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

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

DML_FEATURE_LEVEL_4_1 и выше

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

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_2_1 и выше

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

Требования

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