DML_ACTIVATION_SWISH_OPERATOR_DESC 構造体 (directml.h)

  • [アーティクル]

InputTensor のすべての要素に対してスウィッシュ アクティブ化関数を実行し、結果を OutputTensor の対応する要素に配置します。

f(x) = x * Sigmoid(SigmoidInputScale * x)

この演算子はインプレース実行をサポートしています。つまり、出力テンソルはバインド中に InputTensor のエイリアスを設定できます。

重要

この API は、DirectML スタンドアロン再頒布可能パッケージの一部として使用できます (Microsoft.AI.DirectML バージョン 1.13 以降を参照してください)。 DirectML バージョン履歴も参照してください。

構文

struct DML_ACTIVATION_SWISH_OPERATOR_DESC
{
    const DML_TENSOR_DESC* InputTensor;
    const DML_TENSOR_DESC* OutputTensor;
    FLOAT SigmoidInputScale;
};

メンバー

InputTensor

型: const DML_TENSOR_DESC*

読み込み元の入力テンソル。

OutputTensor

型: const DML_TENSOR_DESC*

結果を書き込む先の出力テンソル。

SigmoidInputScale

型: FLOAT

定数係数。 この値の一般的な既定値は 1.0 です。

可用性

この演算子は、DML_FEATURE_LEVEL_6_2 で導入されました。

Tensor 制約

InputTensor と OutputTensor には、同じ DataType、DimensionCount、およびサイズが必要です。

Tensor のサポート

Tensor 種類 サポートされているディメンション数 サポートされるデータ型
InputTensor 入力 1 から 8 FLOAT32、FLOAT16
OutputTensor 出力 1 から 8 FLOAT32、FLOAT16

要件

   
ヘッダー directml.h