DML_ELEMENT_WISE_ROUND_OPERATOR_DESC結構 （directml.h）

將 InputTensor 的每個元素四捨五入為整數值，並將結果放入 outputTens or的對應元素 。

此運算符支援就地執行，這表示系結期間允許 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：值會四捨五入為零。 這會有效地截斷小數部分。
• 如果 DML_ROUNDING_MODE_TOWARD_INFINITY：值會四捨五入為最接近的整數，其中半途值（例如，0.5）會從零四捨五入（朝正數或負無限大，視值的正負數而定）。

可用性

此運算子是在 DML_FEATURE_LEVEL_2_1中引進的。

Tensor 條件約束

InputTensor 和 OutputTensor 必須具有相同 的 DataType、DimensionCount和 大小。

Tensor 支援

DML_FEATURE_LEVEL_3_0和更新版本

DML_FEATURE_LEVEL_2_1和更新版本

要求