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。

张量约束

InputTensor 和 OutputTensor 必须具有相同的 DataType、 DimensionCount 和 Size。

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