DML_ACTIVATION_PARAMETERIZED_RELU_OPERATOR_DESC構造体 (directml.h)

  • [アーティクル]

InputTensor 内のすべての要素に対してパラメーター化された修正された線形単位 (ReLU) アクティブ化関数を実行し、OutputTensor の対応する要素に結果を配置します。

f(x, slope) = x,         if x >= 0
              slope * x, otherwise

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

構文

struct DML_ACTIVATION_PARAMETERIZED_RELU_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *SlopeTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

メンバー

InputTensor

型: const DML_TENSOR_DESC*

読み取る入力テンソル。

SlopeTensor

型: const DML_TENSOR_DESC*

入力の対応する各値の傾きを含むテンソル。

OutputTensor

型: const DML_TENSOR_DESC*

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

要件

   
Header directml.h

関連項目

DML_ACTIVATION_RELU_OPERATOR_DESC

可用性

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

テンソル制約

InputTensor、 OutputTensor、 SlopeTensor には、 同じ DataType、 DimensionCount、および Sizes が必要です。

Tensor のサポート

DML_FEATURE_LEVEL_5_1以上

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

DML_FEATURE_LEVEL_3_0以上

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

DML_FEATURE_LEVEL_2_0以上

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

DML_FEATURE_LEVEL_1_0以上

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