структура DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC (directml.h)

Статья
08/23/2023

Выполняет функцию естественной активации log-of-softmax для каждого элемента InputTensor, помещая результат в соответствующий элемент OutputTensor.

// Let x_i be the current value in the axis, and j be the total number of elements along that axis.
f(x_i) = ln(exp(x_i) / sum(exp(x_0), ..., exp(x_j)))

Где exp(x) — это функция естественной экспонентации, а ln(x) — естественный логарифм.

Синтаксис

struct DML_ACTIVATION_LOG_SOFTMAX_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

Члены

InputTensor

Тип: const DML_TENSOR_DESC*

Входной тензор для чтения. Этот тензор должен иметь действующий ранг не больше 2. Эффективным рангом тензора является DimensionCount тензора, за исключением крайних левых измерений размера 1. Например, допустимый размер тензора { 1, 1, BatchCount, Width } и эквивалентен тензору размеров { BatchCount, Width }.

OutputTensor

Тип: const DML_TENSOR_DESC*

Выходной тензор для записи результатов.

Доступность

Этот оператор появился в DML_FEATURE_LEVEL_1_0.

Ограничения тензоров

InputTensor и OutputTensor должны иметь одинаковые значения DataType, DimensionCount и Size.

Поддержка тензоров

DML_FEATURE_LEVEL_3_0 и выше

Тензор	Kind	Поддерживаемые счетчики измерений	Поддерживаемые типы данных
InputTensor	Входные данные	От 1 до 8	FLOAT32, FLOAT16
OutputTensor	Выходные данные	От 1 до 8	FLOAT32, FLOAT16

DML_FEATURE_LEVEL_2_0 и выше

Тензор	Kind	Поддерживаемые счетчики измерений	Поддерживаемые типы данных
InputTensor	Входные данные	от 4 до 5	FLOAT32, FLOAT16
OutputTensor	Выходные данные	от 4 до 5	FLOAT32, FLOAT16

DML_FEATURE_LEVEL_1_0 и выше

Тензор	Kind	Поддерживаемые счетчики измерений	Поддерживаемые типы данных
InputTensor	Входные данные	4	FLOAT32, FLOAT16
OutputTensor	Выходные данные	4	FLOAT32, FLOAT16

Требования


Верхняя часть	directml.h

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

структура DML_ACTIVATION_SOFTMAX_OPERATOR_DESC

структура DML_ARGMAX_OPERATOR_DESC

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