Commit df0df33a authored by Carlos Alfredo Yeverino Rodriguez's avatar Carlos Alfredo Yeverino Rodriguez
Browse files

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}, '${}')
Supports Markdown
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