Поделиться через

перечисление 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
Указывает функцию сокращения, которая вычисляет индексы элементов 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

См. также раздел