From b49c2a77d9ed787ad10eb19157f7141253396d83 Mon Sep 17 00:00:00 2001
From: Christian Fuss <chrifuss@freenet.de>
Date: Wed, 3 Jul 2019 23:07:12 +0200
Subject: [PATCH] fixed a problem with inferring size for OneHotLayer, when no
'size' parameter is given
---
src/main/resources/templates/gluon/elements/OneHot.ftl | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/src/main/resources/templates/gluon/elements/OneHot.ftl b/src/main/resources/templates/gluon/elements/OneHot.ftl
index 369bbeaa..66a2fce5 100644
--- a/src/main/resources/templates/gluon/elements/OneHot.ftl
+++ b/src/main/resources/templates/gluon/elements/OneHot.ftl
@@ -1,13 +1,13 @@
<#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=${element.element.outputTypes[0].dimensions[0]})
+ <#include "OutputShape.ftl">
<#elseif mode == "FORWARD_FUNCTION">
${element.name} = self.${element.name}(${input})
<#elseif mode == "PYTHON_INLINE">
- ${element.name} = nd.one_hot(indices=${input}, depth=${size})
+ ${element.name} = nd.one_hot(indices=${input}, depth=${element.element.outputTypes[0].dimensions[0]})
<#elseif mode == "CPP_INLINE">
- vector<float> ${element.name}(${size}, 0);
+ vector<float> ${element.name}(${element.element.outputTypes[0].dimensions[0]}, 0);
${element.name}[${input}[0]] = 1;
</#if>
\ No newline at end of file
--
GitLab