Softmax

Input

x:

Required: yes
Type: float16, float32
Description: input tensor
Restrictions: none

Attribute

axis:

Required: no
Type: int
Description: the axis along which to perform the softmax. Must be in the range [–dim, dim – 1].
Restriction:
vector c<=16384(cs)/24576(ES)

tensor c<=4096

Output

y:

Required: yes
Type: float16, float32
Description: output tensor
Restrictions: none

Parent topic: Supported Caffe Operators