DML_ACTI加值稅ION_LOG_SOFTMAX_OPERATOR_DESC結構 (directml.h)

在 InputTensor的每個元素上執行自然的 softmax 啟用函式，並將結果放入 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 的有效順位 。 張量的有效排名是 tensor 的 DimensionCount ，不包括大小為 1 的最左邊維度。 例如，的 { 1, 1, BatchCount, Width } 張量大小有效，且相當於 大小的 { BatchCount, Width } 張量。

OutputTensor

類型：const DML_TENSOR_DESC*

要寫入結果的輸出張量。

可用性

這個運算子是在 中 DML_FEATURE_LEVEL_1_0 引進。

Tensor 條件約束

InputTensor 和 OutputTensor 必須具有相同 的 DataType、 DimensionCount和 Sizes。

Tensor 支援

DML_FEATURE_LEVEL_3_0和更新版本

張	種類	支援的維度計數	支援的資料類型
InputTensor	輸入	1 到 8	FLOAT32、FLOAT16
OutputTensor	輸出	1 到 8	FLOAT32、FLOAT16

DML_FEATURE_LEVEL_2_0和更新版本

張	種類	支援的維度計數	支援的資料類型
InputTensor	輸入	4 到 5	FLOAT32、FLOAT16
OutputTensor	輸出	4 到 5	FLOAT32、FLOAT16

DML_FEATURE_LEVEL_1_0和更新版本

張	種類	支援的維度計數	支援的資料類型
InputTensor	輸入	4	FLOAT32、FLOAT16
OutputTensor	輸出	4	FLOAT32、FLOAT16

需求

標頭	directml.h

另請參閱

DML_ACTI加值稅ION_SOFTMAX_OPERATOR_DESC結構

DML_ARGMAX_OPERATOR_DESC 結構