softmax layer

parent fdfbe9d5
<#-- This template is not used if the followiing architecture element is an output. See Output.ftl -->
${} = mx.symbol.softmax(data=${element.inputs[0]},
<#assign input = element.inputs[0]>
${} = brew.softmax(model, ${input}, '${}')
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