Atlas系列硬件: Atlas 200l DK A2 开发者套件; Atlas 200l DK 开发者套件（型号：3000）; 加速模块、加速卡、智能小站、服务器、集群

异构计算架构: CANN社区版; CANN商用版

AI框架: MindSpore; TensorFlow框架适配（社区版）; TensorFlow框架适配（商用版）; PyTorch框架适配

分布式训练加速套件: MindSpeed

昇腾推理引擎: MindIE

应用使能MindX: MindX DL; MindX Edge; MindX SDK

全流程开发工具: MindStudio

评分

提单

论坛

小AI

Softmax

输入

x:

是否必填: 必填
数据类型: float16, float32
参数解释: 输入tensor
规格限制: 无

属性

axis:

是否必填: 非必填
数据类型: int
参数解释: softmax的轴，axis取值范围为[-dim,dim-1]
规格限制:
vector c<=16384(cs)/24576(ES)

tensor c<=4096

输出

y:

是否必填: 必填
数据类型: float16,float32
参数解释: 输出Tensor
规格限制: 无

父主题： 支持Caffe算子清单