Compartir a través de

enumeración DML_REDUCE_FUNCTION (directml.h)

  • Artículo

Define constantes que especifican el algoritmo de reducción específico que se va a usar para el operador de reducción de DirectML (como se describe en la estructura de 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
} ;

Constantes

 
DML_REDUCE_FUNCTION_ARGMAX
Indica una función de reducción que calcula los índices de los elementos max de los elementos de tensor de entrada a lo largo del eje especificado, int32 {i j k ..} = maxindex(X Y Z ...).
DML_REDUCE_FUNCTION_ARGMIN
Indica una función de reducción que calcula los índices de los elementos min de los elementos del tensor de entrada a lo largo del eje especificado, int32 {i j k ..} = minindex(X Y Z ...).
DML_REDUCE_FUNCTION_AVERAGE
Indica una función de reducción que calcula la media de los elementos del tensor de entrada a lo largo de los ejes especificados, x = (x1 + x2 + ... + xn) / n.
DML_REDUCE_FUNCTION_L1
Indica una función de reducción que calcula la norma L1 de los elementos del tensor de entrada a lo largo de los ejes especificados, x = |x1| + |x2| + ... + |xn|.
DML_REDUCE_FUNCTION_L2
Indica una función de reducción que calcula la norma L2 de los elementos del tensor de entrada a lo largo de los ejes especificados, x = sqrt(x1^2 + x2^2 + ... + xn^2).
DML_REDUCE_FUNCTION_LOG_SUM
Indica una función de reducción que calcula la suma del registro de los elementos del tensor de entrada a lo largo de los ejes especificados, x = log(x1 + x2 + ... + xn).
DML_REDUCE_FUNCTION_LOG_SUM_EXP
Indica una función de reducción que calcula el exponente de suma de registro de los elementos del tensor de entrada a lo largo de los ejes especificados, x = log(exp(x1) + exp(x2) + ... + exp(xn)).
DML_REDUCE_FUNCTION_MAX
Indica una función de reducción que calcula el máximo de los elementos del tensor de entrada a lo largo de los ejes especificados, x = max(max(max(x1, x2), x3), ..., xn).
DML_REDUCE_FUNCTION_MIN
Indica una función de reducción que calcula el mínimo de los elementos del tensor de entrada a lo largo de los ejes especificados, x = min(min(x1, x2), x3), ..., xn).
DML_REDUCE_FUNCTION_MULTIPLY
Indica una función de reducción que calcula el producto de los elementos del tensor de entrada a lo largo de los ejes especificados, x = (x1 * x2 * ... * xn).
DML_REDUCE_FUNCTION_SUM
Indica una función de reducción que calcula la suma de los elementos del tensor de entrada a lo largo de los ejes especificados, x = (x1 + x2 + ... + xn).
DML_REDUCE_FUNCTION_SUM_SQUARE
Indica una función de reducción que calcula el cuadrado de suma de los elementos del tensor de entrada a lo largo de los ejes especificados, x = x1^2 + x2^2 + ... + xn^2.

Requisitos

   
Encabezado directml.h

Consulte también