DML_FILL_VALUE_SEQUENCE_OPERATOR_DESC 结构 (directml.h)

项目
08/23/2023

用序列填充张量。此运算符执行以下伪代码。

for each coordinate in OutputTensor
    OutputTensor[coordinate] = Value
    Value += Delta
endfor

语法

struct DML_FILL_VALUE_SEQUENCE_OPERATOR_DESC {
  const DML_TENSOR_DESC *OutputTensor;
  DML_TENSOR_DATA_TYPE  ValueDataType;
  DML_SCALAR_UNION      ValueStart;
  DML_SCALAR_UNION      ValueDelta;
};

成员

OutputTensor

类型： const DML_TENSOR_DESC*

要向其写入结果的张量。此张量可能具有任何大小。

ValueDataType

类型： DML_TENSOR_DATA_TYPE

值字段的数据类型，必须与 OutputTensor.DataType 匹配。

ValueStart

类型： DML_SCALAR_UNION

要填充输出中第一个元素的初始值，使用 ValueDataType 确定如何解释字段。

ValueDelta

类型： DML_SCALAR_UNION

向写入的每个元素添加值的步骤， 使用 ValueDataType 确定如何解释字段。

示例

示例 1。 1D 升序步骤

ValueStart = 3
ValueDelta = 2
ValueDataType = DML_TENSOR_DATA_TYPE_FLOAT32

OutputTensor: (Sizes:{1,1,1,3}, DataType:FLOAT32)
    [[[[3, 5, 7]]]]

示例 2。 2D 升序步骤

ValueStart = 10
ValueDelta = -2
ValueDataType = DML_TENSOR_DATA_TYPE_UINT8

OutputTensor: (Sizes:{1,1,2,2}, DataType:UINT8)
    [[[[10, 8],
       [ 6, 4]]]]

可用性

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

Tensor 支持

DML_FEATURE_LEVEL_5_0及更高版本

张	种类	支持的维度计数	支持的数据类型
OutputTensor	输出	1 到 8	FLOAT32、FLOAT16、INT64、INT32、INT16、INT8、UINT64、UINT32、UINT16、UINT8

DML_FEATURE_LEVEL_4_0及更高版本

张	种类	支持的维度计数	支持的数据类型
OutputTensor	输出	1 到 8	FLOAT32、FLOAT16、INT32、INT16、INT8、UINT32、UINT16、UINT8

DML_FEATURE_LEVEL_2_1及更高版本

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

要求


最低受支持的客户端	Windows 10内部版本 20348
最低受支持的服务器	Windows 10内部版本 20348
标头	directml.h

通过