Partager via

énumération DML_REDUCE_FUNCTION (directml.h)

  • Article

Définit des constantes qui spécifient l’algorithme de réduction spécifique à utiliser pour l’opérateur de réduction DirectML (comme décrit par la structure DML_REDUCE_OPERATOR_DESC ).

Syntaxe

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
} ;

Constantes

 
DML_REDUCE_FUNCTION_ARGMAX
Indique une fonction de réduction qui calcule les index des éléments max du tenseur d’entrée le long de l’axe spécifié, int32 {i j k ..} = maxindex(X Y Z ...).
DML_REDUCE_FUNCTION_ARGMIN
Indique une fonction de réduction qui calcule les index des éléments min des éléments du tenseur d’entrée le long de l’axe spécifié, int32 {i j k ..} = minindex(X Y Z ...).
DML_REDUCE_FUNCTION_AVERAGE
Indique une fonction de réduction qui calcule la moyenne des éléments du tenseur d’entrée le long des axes spécifiés, x = (x1 + x2 + ... + xn) / n.
DML_REDUCE_FUNCTION_L1
Indique une fonction de réduction qui calcule la norme L1 des éléments du tenseur d’entrée le long des axes spécifiés, x = |x1| + |x2| + ... + |xn|.
DML_REDUCE_FUNCTION_L2
Indique une fonction de réduction qui calcule la norme L2 des éléments du tenseur d’entrée le long des axes spécifiés, x = sqrt(x1^2 + x2^2 + ... + xn^2).
DML_REDUCE_FUNCTION_LOG_SUM
Indique une fonction de réduction qui calcule la somme du journal des éléments du tenseur d’entrée le long des axes spécifiés, x = log(x1 + x2 + ... + xn).
DML_REDUCE_FUNCTION_LOG_SUM_EXP
Indique une fonction de réduction qui calcule l’exposant de somme du journal des éléments du tenseur d’entrée le long des axes spécifiés, x = log(exp(x1) + exp(x2) + ... + exp(xn)).
DML_REDUCE_FUNCTION_MAX
Indique une fonction de réduction qui calcule la valeur maximale des éléments du tenseur d’entrée le long des axes spécifiés, x = max(max(max(x1, x2), x3), ..., xn).
DML_REDUCE_FUNCTION_MIN
Indique une fonction de réduction qui calcule le min des éléments du tenseur d’entrée le long des axes spécifiés, x = min(min(min(x1, x2), x3), ..., xn).
DML_REDUCE_FUNCTION_MULTIPLY
Indique une fonction de réduction qui calcule le produit des éléments du tenseur d’entrée le long des axes spécifiés, x = (x1 * x2 * ... * xn).
DML_REDUCE_FUNCTION_SUM
Indique une fonction de réduction qui calcule la somme des éléments du tenseur d’entrée le long des axes spécifiés, x = (x1 + x2 + ... + xn).
DML_REDUCE_FUNCTION_SUM_SQUARE
Indique une fonction de réduction qui calcule le carré de somme des éléments du tenseur d’entrée le long des axes spécifiés, x = x1^2 + x2^2 + ... + xn^2.

Spécifications

   
En-tête directml.h

Voir aussi