directml.h) (DML_ACTIVATION_HARDMAX_OPERATOR_DESC 结构
对 InputTensor 的每个元素执行 hardmax 函数，将结果放入 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 表示给定输入批处理中每个层的所有其他) 值。 输入是大小为 (batch_size x input_feature_dimensions) 的二维张量 (张量) 。 输出张量具有相同的形状，并且包含相应输入的 hardmax 值。
可用性
此运算符是在 中 DML_FEATURE_LEVEL_1_0引入的。
张量约束
InputTensor 和 OutputTensor 必须具有相同的 DataType、 DimensionCount 和 Size。
张量支持
DML_FEATURE_LEVEL_3_0 及更高版本
| 张 | 种类 | 支持的维度计数 | 支持的数据类型 |
|---|---|---|---|
| InputTensor | 输入 | 1 到 8 | FLOAT32、FLOAT16 |
| OutputTensor | 输出 | 1 到 8 | FLOAT32、FLOAT16 |
DML_FEATURE_LEVEL_2_0及更高版本
| 张 | 种类 | 支持的维度计数 | 支持的数据类型 |
|---|---|---|---|
| InputTensor | 输入 | 4 到 5 | FLOAT32、FLOAT16 |
| OutputTensor | 输出 | 4 到 5 | FLOAT32、FLOAT16 |
DML_FEATURE_LEVEL_1_0 及更高版本
| 张 | 种类 | 支持的维度计数 | 支持的数据类型 |
|---|---|---|---|
| InputTensor | 输入 | 4 | FLOAT32、FLOAT16 |
| OutputTensor | 输出 | 4 | FLOAT32、FLOAT16 |
要求
| 要求 | 值 |
|---|---|
| Header | directml.h |