softmax layer

parent fdfbe9d5
<#-- This template is not used if the followiing architecture element is an output. See Output.ftl -->
${element.name} = mx.symbol.softmax(data=${element.inputs[0]},
axis=1,
name="${element.name}")
<#assign input = element.inputs[0]>
${element.name} = brew.softmax(model, ${input}, '${element.name}')
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment