DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC構造体 (directml.h)

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

指定した InfinityModeに応じて、IEEE-754 -inf、inf、またはその両方の InputTensor の各要素をチェックし、結果 (true の場合は 1、false の場合は 0) を OutputTensorの対応する要素配置します。

f(x) = isinf(x) && (
       (x > 0 && InfinityMode == DML_IS_INFINITY_MODE_POSITIVE) ||
       (x < 0 && InfinityMode == DML_IS_INFINITY_MODE_NEGATIVE) ||
                 InfinityMode == DML_IS_INFINITY_MODE_EITHER)

構文

struct DML_ELEMENT_WISE_IS_INFINITY_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *OutputTensor;
  DML_IS_INFINITY_MODE  InfinityMode;
};

メンバーズ

InputTensor

型: const DML_TENSOR_DESC*

読み取る入力テンソル。

OutputTensor

型: const DML_TENSOR_DESC*

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

InfinityMode

型: DML_IS_INFINITY_MODE

チェックする無限大の符号を決定する DML_IS_INFINITY_MODE。

DML_IS_INFINITY_MODE_EITHER場合、要素が -inf または inf の場合は 1 が返され、それ以外の場合は 0 が返されます。
DML_IS_INFINITY_MODE_POSITIVE場合、要素が inf の場合は 1 が返され、それ以外の場合は 0 が返されます。
'DML_IS_INFINITY_MODE_NEGATIVE場合、要素が -inf の場合は 1 が返され、それ以外の場合は 0 が返されます。

備考

可用性

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

Tensor 制約

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

Tensor のサポート

DML_FEATURE_LEVEL_3_0以上

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

DML_FEATURE_LEVEL_2_1以降

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

必要条件

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

次の方法で共有