通过

DML_ELEMENT_WISE_ROUND_OPERATOR_DESC结构（directml.h）

  • 项目

将 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：值将舍入为零。 这实际上截断了小数部分。
  • 如果 DML_ROUNDING_MODE_TOWARD_INFINITY：将值舍入到最接近的整数，中间值（例如，0.5）从零舍入（朝正或负无穷大，具体取决于值的符号）。

可用性

此运算符是在 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