下载
中文
注册

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
  • 规格限制: 无