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。

张	种类	支持的维度计数	支持的数据类型
InputTensor	输入	1 到 8	FLOAT32、FLOAT16
OutputTensor	输出	1 到 8	FLOAT32、FLOAT16

张	种类	支持的维度计数	支持的数据类型
InputTensor	输入	4 到 5	FLOAT32、FLOAT16
OutputTensor	输出	4 到 5	FLOAT32、FLOAT16

张	种类	支持的维度计数	支持的数据类型
InputTensor	输入	4	FLOAT32、FLOAT16
OutputTensor	输出	4	FLOAT32、FLOAT16

要求	值
Header	directml.h