DML_REDUCE_FUNCTION (directml.h) 列舉

定義常數,指定要用於 DirectML 歸納運算子的特定縮減演算法 (,如 DML_REDUCE_OPERATOR_DESC 結構) 所述。

Syntax

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
表示縮減函式,計算輸入張量專案沿著指定座標軸的 min 元素索引 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 = log (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

另請參閱