Commit 5573c542 authored by Christian Fuß's avatar Christian Fuß
Browse files

Made 'size' parameter for OneHot Layer optional.

parent 5999ce8e
Pipeline #153414 failed with stages
in 2 minutes and 12 seconds
<#assign input = element.inputs[0]>
<#assign size = element.size>
<#if mode == "ARCHITECTURE_DEFINITION">
self.${element.name} = OneHot(size=${size})
<#include "OutputShape.ftl">
self.${element.name} = OneHot(size=${size})
<#include "OutputShape.ftl">
</#if>
<#if mode == "FORWARD_FUNCTION">
${element.name} = self.${element.name}(${input})
......
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