次の方法で共有

DML_ACTIVATION_CELU_OPERATOR_DESC構造体 (directml.h)

  • [アーティクル]

InputTensor 内のすべての要素に対して連続して異なる指数線形単位 (CELU) アクティブ化関数を実行し、OutputTensor の対応する要素に結果を配置します。

f(x) = max(0, x) + min(0, Alpha * (exp(x / Alpha) - 1));

各値の説明:

  • exp(x) は自然な指数関数です
  • max(a,b) は、2 つの値 a,b のうち大きい方を返します
  • min(a,b) は、2 つの値 a,b のうち小さい方を返します

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

構文

struct DML_ACTIVATION_CELU_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Alpha;
};

メンバー

InputTensor

型: const DML_TENSOR_DESC*

読み取る入力テンソル。

OutputTensor

型: const DML_TENSOR_DESC*

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

Alpha

型: FLOAT

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

可用性

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

テンソル制約

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

Tensor のサポート

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

要件

   
サポートされている最小のクライアント Windows 10 ビルド 20348
サポートされている最小のサーバー Windows 10 ビルド 20348
Header directml.h