DML_ELEMENT_WISE_THRESHOLD_OPERATOR_DESC 構造体 (directml.h)

[アーティクル]
08/23/2023

指定されたしきい値 Min を下回る InputTensor のすべての要素を Min に置き換えます。結果は OutputTensor の対応する要素に配置されます。

f(x) = max(x, Min)

ここで、max(a,b) は、2 つの値 a,b の大きい方を返します。

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

構文

struct DML_ELEMENT_WISE_THRESHOLD_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  const DML_SCALE_BIAS  *ScaleBias;
  FLOAT                 Min;
};

メンバー

InputTensor

型: const DML_TENSOR_DESC*

読み取る入力テンソル。

OutputTensor

型: const DML_TENSOR_DESC*

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

ScaleBias

型: _Maybenull_ const DML_SCALE_BIAS*

入力に適用するスケールとバイアス (省略可能)。存在する場合、この演算子を計算する前に、各入力要素に関数g(x) = x * scale + biasを適用する効果があります。

Min

型: FLOAT

最小値。この最小値の下で、演算子は値を Min に置き換えます。

可用性

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

テンソル制約

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

Tensor のサポート

DML_FEATURE_LEVEL_3_0以上

テンソル	種類	サポートされているディメンション数	サポートされるデータ型
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_2_1以上

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

DML_FEATURE_LEVEL_1_0以降

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

要件

要件	値
Header	directml.h