Share via

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

  • Статья

Выполняет функцию hardmax для каждого элемента InputTensor, помещая результат в соответствующий элемент OutputTensor.

Оператор вычисляет hardmax (1 для первого вхождения наибольшего значения в слое и 0 для всех остальных значений) каждой строки в заданном входе.

Синтаксис

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

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

Тензор для считывания входных данных. Этот тензор должен иметь действующий ранг не больше 2. Эффективный ранг тензора — это DimensionCount тензора, за исключением крайних левых измерений размера 1. Например, размер тензора является допустимым { 1, 1, BatchCount, Width } и эквивалентен тензору размеров { BatchCount, Width }.

OutputTensor

Тип: const DML_TENSOR_DESC*

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

Комментарии

Оператор вычисляет значения hardmax (1 для первого максимального значения и 0 для всех остальных) для каждого слоя в пакете заданных входных данных. Входные данные являются объемным тензором (tensor) размера (batch_size x input_feature_dimensions). Выходной тензор имеет ту же форму и содержит значения hardmax соответствующих входных данных.

Доступность

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

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

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

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

DML_FEATURE_LEVEL_3_0 и выше

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

DML_FEATURE_LEVEL_2_0 и выше

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

DML_FEATURE_LEVEL_1_0 и выше

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

Требования

Требование Значение
Заголовок directml.h

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

структура DML_ARGMAX_OPERATOR_DESC