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 |