directml.h) (DML_FILL_VALUE_CONSTANT_OPERATOR_DESC 结构
使用给定的常量 Value 填充张量。 此运算符执行以下伪代码。
for each coordinate in OutputTensor
OutputTensor[coordinate] = Value
endfor
语法
struct DML_FILL_VALUE_CONSTANT_OPERATOR_DESC {
const DML_TENSOR_DESC *OutputTensor;
DML_TENSOR_DATA_TYPE ValueDataType;
DML_SCALAR_UNION Value;
};
成员
OutputTensor
类型: const DML_TENSOR_DESC*
将结果写入到的张量。 此张量可以具有任何大小。
ValueDataType
“值”字段的数据类型,必须与 OutputTensor.DataType 匹配。
Value
类型: DML_SCALAR_UNION
用于填充输出的常量值,其中 ValueDataType 确定如何解释字段。
示例
Value = 13.0
OutputTensor: (Sizes:{1,1,2,4}, DataType:FLOAT32)
[[[[13.0f, 13.0f, 13.0f, 13.0f],
[13.0f, 13.0f, 13.0f, 13.0f]]]]
可用性
此运算符是在 中 DML_FEATURE_LEVEL_2_1
引入的。
张量支持
DML_FEATURE_LEVEL_4_1 及更高版本
张 | 种类 | 支持的维度计数 | 支持的数据类型 |
---|---|---|---|
OutputTensor | 输出 | 1 到 8 | FLOAT64、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 |