三亩地 三亩地SAN MU DI · CODE DIARY
ARTICLE DETAIL

日记详情

真实记录编程学习的某一天,欢迎挑你感兴趣的翻一翻。

CANN/ops-math Fills填充算子

CANN/ops-math Fills填充算子

Fills

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math

产品支持情况

产品是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品
Atlas 推理系列产品
Atlas 训练系列产品

功能说明

  • 算子功能:填充输入tensor所有元素为指定值。

  • 计算公式:

$$out_i=value$$

参数说明

参数名输入/输出/属性描述数据类型数据格式
x输入待进行Fills计算的入参。FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、INT32、INT64ND
value输入属性待进行Fills填充值,公式中的value。FLOATND
y输出待进行Fills计算的出参,公式中的out_i。FLOAT、FLOAT16、BFLOAT16、INT8、UINT8、INT32、INT64ND

约束说明

调用说明

调用方式调用样例说明
图模式调用test_geir_fills通过算子IR构图方式调用Fills算子。

【免费下载链接】ops-math本项目是CANN提供的数学类基础计算算子库,实现网络在NPU上加速计算。项目地址: https://gitcode.com/cann/ops-math

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

← 返回列表