Softmax.ftl 224 Bytes
Newer Older
1
<#-- This template is not used if the following architecture element is an output. See Output.ftl -->
Nicola Gatto's avatar
Nicola Gatto committed
2
<#assign input = element.inputs[0]>
3
4
<#if mode == "FORWARD_FUNCTION">
        ${element.name} = F.softmax(${input})
Nicola Gatto's avatar
Nicola Gatto committed
5
</#if>