updating the signature for the Embedding() macro

putraxor · Apr 15, 2016 · 3f48557 · 3f48557
1 parent 60f4b14
commit 3f48557
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/Source/CNTK/BrainScript/CNTKCoreLib/CNTK.core.bs b/Source/CNTK/BrainScript/CNTKCoreLib/CNTK.core.bs
@@ -151,10 +151,10 @@ SBFF(in, rows, cols) = [ Eh = Sigmoid(BFF(in, rows, cols).z) ]
 MeanVarNorm(feat) = PerDimMeanVarNormalization(feat, Mean(feat), InvStdDev(feat)) 
 LogPrior(labels) = Log(Mean(labels)) 
 
-Embedding (input, inputDim, embeddingDim, initFrom='fromFile'/*|gaussian|uniform*/, embeddingPath = '', sparseInput = false, learningRateWeight = 0.0) = [
-    embedding = Transpose (LearnableParameter (inputDim, embeddingDim, learningRateMultiplier = learningRateWeight, init = initFrom, initFromFilePath = embeddingPath))          
+Embedding (embeddingDim, input, inputDim=input.dim, initFrom='fromFile'/*|gaussian|uniform*/, embeddingPath = '', sparseInput = false, learningRateWeight = 0.0) = [
+    embedding = Transpose (LearnableParameter (inputDim, embeddingDim, learningRateMultiplier = learningRateWeight, init = initFrom, initFromFilePath = embeddingPath))
     lookup = if sparseInput then embedding * input
-             else GatherPacked (input, embedding) 
+             else GatherPacked (input, embedding)
 ].lookup
 
 ##############################################################################