DML_ACTIVATION_SHRINK_OPERATOR_DESC 構造体 (directml.h)

[アーティクル]
02/28/2024

InputTensor 内のすべての要素に対して圧縮アクティブ化関数を実行し、OutputTensor の対応する要素に結果を配置します。

f(x) = x - Bias, if x > Threshold
       x + Bias, if x < -Threshold
       0,        otherwise

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

構文

struct DML_ACTIVATION_SHRINK_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  FLOAT                 Bias;
  FLOAT                 Threshold;
};

メンバー

InputTensor

型: const DML_TENSOR_DESC*

読み取る入力テンソル。

OutputTensor

型: const DML_TENSOR_DESC*

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

Bias

型: FLOAT

バイアスの値。この値の一般的な既定値は 0.0 です。

Threshold

型: FLOAT

しきい値の値。この値の一般的な既定値は 0.5 です。

注釈

可用性

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

テンソル制約

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

Tensor のサポート

DML_FEATURE_LEVEL_5_1以降

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

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

要件

要件	値
サポートされている最小のクライアント	Windows 10バージョン 2004 (10.0;ビルド 19041)
サポートされている最小のサーバー	Windows Server バージョン 2004 (10.0;ビルド 19041)
Header	directml.h

次の方法で共有