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 --> <#-- 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]}, <#assign input = element.inputs[0]>
axis=1, ${element.name} = brew.softmax(model, ${input}, '${element.name}')
name="${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