DML_MAX_UNPOOLING_OPERATOR_DESC 结构 (directml.h)

  • 项目

反转 max-pooling 操作 (请参阅 DML_MAX_POOLING_OPERATOR1_DESC，了解) 的详细信息，方法是根据 IndexesTensor 中提供的索引值，使用输入张量 InputTensor 中的值填充输出张量 OutputTensor。 进程未触及的输出张量中的元素将保留零个值。

语法

struct DML_MAX_UNPOOLING_OPERATOR_DESC {
  const DML_TENSOR_DESC *InputTensor;
  const DML_TENSOR_DESC *IndicesTensor;
  const DML_TENSOR_DESC *OutputTensor;
};

成员

InputTensor

类型： const DML_TENSOR_DESC*

大小的{ Batch, Channel, Height, Width }输入张量。 张量值是从 max-pooling 操作的 OutputTensor 中的值获取的。

IndicesTensor

类型： const DML_TENSOR_DESC*

输入张量 InputTensor 中给定的值的索引到输出张量 OutputTensor。 这些索引值从零开始，并将输出张量视为连续的一维数组。 InputTensor 和 IndicesTensor 具有相同的张量大小。 张量值是从 max-pooling 操作的 OutputIndicesTensor 中获取的。

OutputTensor

类型： const DML_TENSOR_DESC*

与输入张量具有相同维度数的输出张量。

可用性

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

张量约束

InputTensor 和 OutputTensor 必须具有相同的 数据类型。

张量支持

DML_FEATURE_LEVEL_5_0 及更高版本

张 种类 支持的维度计数 支持的数据类型
InputTensor 输入 4 FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8
IndicesTensor 输入 4 UINT64、UINT32
OutputTensor 输出 4 FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8

DML_FEATURE_LEVEL_3_0 及更高版本

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

要求

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