структура 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 |