DML_ACTIVATION_THRESHOLDED_RELU_OPERATOR_DESC 構造体 (directml.h)
InputTensor 内のすべての要素に対してしきい値修正された線形単位 (ReLU) アクティブ化関数を実行し、OutputTensor の対応する要素に結果を配置します。
f(x) = x, if x > Alpha
0, otherwise
この演算子はインプレース実行をサポートしています。つまり、バインド中に InputTensor のエイリアスを出力テンソルに許可します。
構文
struct DML_ACTIVATION_THRESHOLDED_RELU_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_1_0導入されました。
テンソル制約
InputTensor と OutputTensor には、同じ DataType、 DimensionCount、および サイズが必要です。
Tensor のサポート
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 |
DML_FEATURE_LEVEL_1_0以降
| テンソル | 種類 | サポートされているディメンション数 | サポートされるデータ型 |
|---|---|---|---|
| InputTensor | 入力 | 4 | FLOAT32、FLOAT16 |
| OutputTensor | 出力 | 4 | FLOAT32、FLOAT16 |
要件
| Header | directml.h |