OneHot.ftl 299 Bytes
Newer Older
Christian Fuß's avatar
Christian Fuß committed
1
2
3
<#assign input = element.inputs[0]>
<#assign size = element.size>
<#if mode == "ARCHITECTURE_DEFINITION">
4
5
        self.${element.name} = OneHot(size=${size})
        <#include "OutputShape.ftl">
Christian Fuß's avatar
Christian Fuß committed
6
7
8
9
</#if>
<#if mode == "FORWARD_FUNCTION">
        ${element.name} = self.${element.name}(${input})
</#if>