DML_ACTIVATION_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 以下である必要があります。 テンソルの有効ランクは、サイズ 1 の左端のディメンションを除くテンソルの DimensionCount です。 たとえば、 の { 1, 1, BatchCount, Width }
テンソル サイズは有効であり、サイズ { BatchCount, Width }
のテンソルと同じです。
OutputTensor
型: const DML_TENSOR_DESC*
結果を書き込む出力テンソル。
可用性
この演算子は、 で DML_FEATURE_LEVEL_1_0
導入されました。
テンソル制約
InputTensor と OutputTensor には、同じ DataType、 DimensionCount、および サイズが必要です。
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 |
要件
Header | directml.h |