Fill
产品支持情况
| 产品 | 是否支持 |
|---|---|
| Atlas A2 训练系列产品/Atlas A2 推理系列产品 | √ |
功能说明
-
算子功能:实现张量的填充功能,将张量中的所有元素填充为指定的统一值。
-
计算公式: 对于任意索引位置
(i, j, ...),执行fill(T, v)后:
∀(i,j,… ),T[i,j,… ]=v\forall (i, j, \dots), \quad T[i, j, \dots] = v
参数说明
| 参数名 | 输入/输出/属性 | 描述 | 数据类型 | 数据格式 |
|---|---|---|---|---|
| x | 输入 | 待进行fill算子计算的入参,公式中的T | FLOAT16、FLOAT、INT16、INT32、BF16 | ND |
| fill_value | 输入 | 用于填充的标量值,公式中的v | FLOAT16、FLOAT、INT16、INT32、BF16 | ND |
| x | 输出 | 执行fill算子计算后的出参,公式中的T' | FLOAT16、FLOAT、INT16、INT32、BF16 | ND |
约束说明
无
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_fill_v2.cpp | 通过[test_aclnn_s_where]接口方式调用Fill算子。 |