通过

DML_MAX_UNPOOLING_OPERATOR_DESC结构（directml.h）

根据 IndexesTensor中提供的索引值，通过填充输出张量 OutputTensor，来反转最大池操作（请参阅 DML_MAX_POOLING_OPERATOR1_DESC 了解详细信息），方法是使用输入张量 InputTensor中的值。 此过程未触及的输出张量中的元素保留为零值。

语法

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 }。 张量值是从最大池操作 OutputTensor 中的值获取的。

IndicesTensor

类型：const DML_TENSOR_DESC*

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

OutputTensor

类型：const DML_TENSOR_DESC*

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

可用性

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

Tensor 约束

InputTensor 和 OutputTensor 必须具有相同 DataType。

Tensor 支持

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
  • Last updated on