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

SoftmaxV2

📄 查看源码

产品支持情况

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

功能说明

  • 算子功能:对输入张量计算Softmax值。

  • 计算公式:

    Softmax(xi)=exi∑j=0n−1exjSoftmax\left ( {{x}_{i}} \right )= {\frac {{e}^{{x}_{i}}} {\sum ^{n-1}_{j=0} {{e}^{{x}_{j}}}}}

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 待进行Softmax计算的入参,公式中的x。 FLOAT,FLOAT16,BFLOAT16 ND
axes 输入 需要做Softmax的轴。只支持1维,默认-1。 INT32_ARRAY -
half_to_float 输入 可选的布尔值。此参数决定当输入数据类型为float16时,是否将输出数据类型转换为float32。默认值为false。值为false时,输入输出数据类型要求一致。如果为true且输入数据类型为float16,则输出数据类型应为float32。 BOOL -
y 输出 公式中的输出张量 FLOAT,FLOAT16,BFLOAT16 ND

约束说明

调用说明

调用方式 样例代码 说明
aclnn接口 test_aclnn_softmax_v2 通过aclnnSoftmax接口方式调用SoftmaxV2算子。