文件最后提交记录最后更新时间
13 小时前
17 天前
17 天前
17 天前
17 天前
17 天前
5 个月前
20 小时前
README

Glu

产品支持情况

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

功能说明

  • 接口功能:GLU是一个门控线性单元函数,它将输入张量沿着指定的维度dim平均分成两个张量,并将其前部分张量与后部分张量的Sigmoid函数输出的结果逐元素相乘。

  • 计算公式:

    GLU(a,b)=a⊗σ(b)GLU(a,b)=a \otimes \sigma(b)

    a表示的是输入张量根据指定dim进行均分后的前部分张量,b表示后半部分张量。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 输入一个张量。 DT_BF16、FLOAT16、DT_FLOAT ND
self 输入 待进行GLU计算的入参,将该输入张量沿着指定的维度dim平均分成公式中的两个张量a,b。 DOUBLE、FLOAT、FLOAT16、BFLOAT16 ND
dim 输入 表示要拆分输入self的维度。 INT64 -
out 输出 计算出参。 DOUBLE、FLOAT、FLOAT16、BFLOAT16 ND

约束说明

调用说明

调用方式 调用样例 说明
aclnn调用 test_aclnn_glu 通过aclnnGlu接口方式调用GLU算子。