DML_REDUCE_FUNCTION 枚举 (directml.h)

定义常量，这些常量指定用于 DirectML reduce 运算符 (的特定缩减算法，如 DML_REDUCE_OPERATOR_DESC 结构) 所述。

语法

typedef enum DML_REDUCE_FUNCTION {
  DML_REDUCE_FUNCTION_ARGMAX,
  DML_REDUCE_FUNCTION_ARGMIN,
  DML_REDUCE_FUNCTION_AVERAGE,
  DML_REDUCE_FUNCTION_L1,
  DML_REDUCE_FUNCTION_L2,
  DML_REDUCE_FUNCTION_LOG_SUM,
  DML_REDUCE_FUNCTION_LOG_SUM_EXP,
  DML_REDUCE_FUNCTION_MAX,
  DML_REDUCE_FUNCTION_MIN,
  DML_REDUCE_FUNCTION_MULTIPLY,
  DML_REDUCE_FUNCTION_SUM,
  DML_REDUCE_FUNCTION_SUM_SQUARE
} ;

常量

DML_REDUCE_FUNCTION_ARGMAX 指示一个缩减函数，该函数计算指定轴上输入张量元素的最大元素索引，即 int32 {i j k ..} = maxindex (X Y Z ...) 。
DML_REDUCE_FUNCTION_ARGMIN 指示一个缩减函数，该函数计算输入张量元素沿指定轴 int32 {i j k ..} 的最小元素的索引 = minindex (X Y Z ...) 。
DML_REDUCE_FUNCTION_AVERAGE 指示沿指定轴计算输入张量元素的平均值的缩减函数，x = (x1 + x2 + ... + xn) / n。
DML_REDUCE_FUNCTION_L1 指示沿指定轴计算输入张量元素的 L1 范数的缩减函数，x = |x1|+ |x2|+ ... + |xn|。
DML_REDUCE_FUNCTION_L2 指示沿指定轴计算输入张量元素的 L2 范数的缩减函数，x = sqrt (x1^2 + x2^2 + ... + xn^2) 。
DML_REDUCE_FUNCTION_LOG_SUM 指示沿指定轴计算输入张量元素的对数之和的缩减函数，x = 对数 (x1 + x2 + ... + xn) 。
DML_REDUCE_FUNCTION_LOG_SUM_EXP 指示沿指定轴计算输入张量元素的对数和指数的缩减函数，x = log (exp (x1) + exp (x2) + ... + exp (xn) ) 。
DML_REDUCE_FUNCTION_MAX 指示沿指定轴计算输入张量元素的最大值的缩减函数，x = max (max (max (x1、x2) 、x3) 、...、xn) 。
DML_REDUCE_FUNCTION_MIN 指示沿指定轴计算输入张量元素的最小值的缩减函数，x = min (min (min (x1、x2) 、x3) 、...、xn) 。
DML_REDUCE_FUNCTION_MULTIPLY 指示沿指定轴计算输入张量元素的乘积的缩减函数，x = (x1 * x2 * ... * xn) 。
DML_REDUCE_FUNCTION_SUM 指示沿指定轴计算输入张量元素的总和的缩减函数，x = (x1 + x2 + ... + xn) 。
DML_REDUCE_FUNCTION_SUM_SQUARE 指示一个缩减函数，该函数计算指定轴上输入张量元素的平方和，x = x1^2 + x2^2 + ... + xn^2。

要求

标头	directml.h

请参阅

• DML_REDUCE_OPERATOR_DESC