Fill
贡献说明
| 贡献者 | 贡献方 | 贡献算子 | 贡献时间 | 贡献内容 |
|---|---|---|---|---|
| skywang2 | 个人开发者 | Fill | 2025/12/31 | 新增Fill算子 |
支持的产品型号
- Atlas A2训练系列产品
算子描述
-
功能描述
Fill算子创建一个形状由输入dims指定的张量，并用标量值value填充所有元素。该算子常用于初始化张量为特定值。
-
原型信息
| 算子类型(OpType) | Fill | |||
|---|---|---|---|---|
| name | Type | data type | format | |
| 算子输入 | ||||
| dims | attr_tuple | int64 | - | |
| 算子输入 | ||||
| value | scalar | float32,float16,bfloat16,int8,bool,int64,int32 | - | |
| 算子输出 | y | tensor | float32,float16,bfloat16,int8,bool,int64,int32 | ND |
| 核函数名 | Fill | |||
-
约束与限制
dims的数据类型支持INT64。
value的数据类型支持FLOAT16、FLOAT32、INT8、INT32、INT64，BOOL、BFLOAT16。
y的数据类型支持FLOAT16、FLOAT32、INT8、INT32、INT64，BOOL、BFLOAT16。，数据格式只支持ND。
环境要求
编译运行此样例前，请参考《CANN软件安装指南》完成开发运行环境的部署。
运行验证
测试命令调用方式：build.sh
| 目录 | 描述 |
|---|---|
| test_aclnn_fill.cpp | 通过aclnn调用的方式调用Fill算子。 |