文件最后提交记录最后更新时间
6 个月前
7 个月前
6 个月前
2 个月前
7 个月前
6 个月前
7 个月前
4 个月前
README

Sign

产品支持情况

产品 是否支持
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品

功能说明

  • 算子功能:对输入逐元素取符号,正数为1,负数为−1,0为0。

  • 计算公式:

    outi={1if inputi>00if inputi=0−1if inputi<0\text{out}_{i}= \begin{cases} 1 & \text{if } \text{input}_{i}>0 \\[4pt] 0 & \text{if } \text{input}_{i}=0 \\[4pt] -1 & \text{if } \text{input}_{i}<0 \end{cases}

参数说明:

参数名 输入/输出 描述 使用说明 数据类型 数据格式 维度(shape) 非连续Tensor
input 输入 待取符号的Tensor。 支持空Tensor。 FLOAT、FLOAT16、BFLOAT16、INT32 ND 1-8
output 输出 与input同形状的符号结果。 数据类型、shape需与input一致。 FLOAT、FLOAT16、BFLOAT16、INT32 同input ND

算子使用

使用该算子前,请参考社区版CANN开发套件包安装文档完成开发运行环境的部署。

编译部署

  • 进入到仓库目录

    cd ${git_clone_path}/ops-math
    
  • 执行编译

    bash build.sh --pkg --experimental --soc=ascend910b --ops=sign
    
  • 部署算子包

    ./build_out/cann-ops-<vendor_name>-linux.<arch>.run
    

算子调用

  • 执行调用

    bash build.sh --run_example sign eager cust --vendor_name=custom
    

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_sign 通过aclnnSign接口方式调用Sign算子。

贡献说明

贡献者 贡献方 贡献算子 贡献时间 贡献内容
hth810 个人开发者 Sign 2025/12/12 Sign算子适配开源仓