DML_RESAMPLE_OPERATOR_DESC 结构 (directml.h)

项目
01/10/2024

使用比例系数计算目标张量大小，将元素从源重新采样到目标张量。可以使用线性或近邻内插模式。运算符支持跨多个维度的内插，而不仅仅是 2D。因此，可以保留相同的空间大小，但可以跨通道或跨批内插。

语法

struct DML_RESAMPLE_OPERATOR_DESC {
  const DML_TENSOR_DESC  *InputTensor;
  const DML_TENSOR_DESC  *OutputTensor;
  DML_INTERPOLATION_MODE InterpolationMode;
  UINT                   ScaleCount;
  const FLOAT            *Scales;
};

成员

InputTensor

类型： const DML_TENSOR_DESC*

包含输入数据的张量。

OutputTensor

类型： const DML_TENSOR_DESC*

要向其写入输出数据的张量。

InterpolationMode

类型： DML_INTERPOLATION_MODE

此字段确定用于选择输出像素的内插类型。

DML_INTERPOLATION_MODE_NEAREST_NEIGHBOR。使用 Nearest Neighbor 算法，该算法为每个输出元素选择最靠近相应像素中心的输入元素。
DML_INTERPOLATION_MODE_LINEAR。使用 线性内插 算法，该算法通过计算每个维度最近 2 个相邻输入元素的加权平均值来计算输出元素。重采样支持最多 4 个维度 (四维线性) ，其中加权平均值根据每个输出元素的总共 16 个输入元素进行计算。

ScaleCount

类型： UINT

数组 Scales 中指向的值的数目。此值必须与 InputTensor 和 OutputTensor 的维度计数匹配。

Scales

类型：_Field_size_ (ScaleCount) const FLOAT*

重新采样输入时要应用的缩放比例，其中缩放 > 1 纵向扩展图像， < 缩放 1 缩放该维度的图像。请注意，刻度不需要完全 OutputSize / InputSize为。如果缩放后的输入大于输出绑定，则将其裁剪为输出大小。另一方面，如果缩放后的输入小于输出绑定，则会固定输出边缘。

注解

在中DML_FEATURE_LEVEL_2_1引入了此运算符的较新版本（DML_RESAMPLE1_OPERATOR_DESC）。

可用性

此运算符是在中引入的 DML_FEATURE_LEVEL_2_0。

张量约束

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

Tensor 支持

DML_FEATURE_LEVEL_6_2及更高版本

张	种类	支持的维度计数	支持的数据类型
InputTensor	输入	1 到 4	FLOAT32、FLOAT16、INT8、UINT8
OutputTensor	输出	1 到 4	FLOAT32、FLOAT16、INT8、UINT8

DML_FEATURE_LEVEL_5_1 及更高版本

张	种类	支持的维度计数	支持的数据类型
InputTensor	输入	1 到 4	FLOAT32、FLOAT16
OutputTensor	输出	1 到 4	FLOAT32、FLOAT16

DML_FEATURE_LEVEL_2_0及更高版本

张	种类	支持的维度计数	支持的数据类型
InputTensor	输入	4	FLOAT32、FLOAT16
OutputTensor	输出	4	FLOAT32、FLOAT16

要求


最低受支持的客户端	Windows 10版本 2004 (10.0;内部版本 19041)
最低受支持的服务器	Windows Server 版本 2004 (10.0;内部版本 19041)
标头	directml.h

通过