DML_ELEMENT_WISE_ROUND_OPERATOR_DESC構造体 (directml.h)

[アーティクル]
11/12/2024

InputTensor の各要素を整数値に丸め、結果を OutputTensorの対応する要素配置します。

この演算子はインプレース実行をサポートしています。つまり、バインド時に OutputTensor が InputTensor エイリアスを作成できます。

構文

struct DML_ELEMENT_WISE_ROUND_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_ROUNDING_MODE     RoundingMode;
};

メンバーズ

InputTensor

型: const DML_TENSOR_DESC*

読み取る入力テンソル。

OutputTensor

型: const DML_TENSOR_DESC*

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

RoundingMode

型: DML_ROUNDING_MODE

丸める方向を決定する DML_ROUNDING_MODE。

DML_ROUNDING_MODE_HALVES_TO_NEAREST_EVENの場合: 値は最も近い整数に丸められ、中間値 (0.5 など) は最も近い偶数の整数に丸められます。
DML_ROUNDING_MODE_TOWARD_ZEROの場合: 値は 0 に丸められます。これにより、小数部が実質的に切り捨てられます。
DML_ROUNDING_MODE_TOWARD_INFINITYの場合: 値が最も近い整数に丸められ、中途半端な値 (例: 0.5) が 0 から丸められます (値の符号に応じて、正または負の無限大に向かって)。

可用性

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

Tensor 制約

InputTensor と OutputTensor 、DataType、DimensionCount、およびサイズが同じである必要があります。

Tensor のサポート

DML_FEATURE_LEVEL_3_0以上

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

DML_FEATURE_LEVEL_2_1以降

テンソル	種類	サポートされているディメンション数	サポートされているデータ型
InputTensor	インプット	4	FLOAT32、FLOAT16
OutputTensor	アウトプット	4	FLOAT32、FLOAT16

必要条件

要件	価値
サポートされる最小クライアント	Windows 10 ビルド 20348
サポートされる最小サーバー	Windows 10 ビルド 20348
ヘッダー	directml.h

次の方法で共有