OneHot.ftl 176 Bytes
Newer Older
Christian Fuß's avatar
Christian Fuß committed
1
<#assign input = element.inputs[0]>
2
<#assign size = element.size?c>
3
4
<#if mode == "FORWARD_FUNCTION">
        ${element.name} = F.one_hot(indices=${input}, depth=${size})
5
</#if>