Merge pull request #1886 from pluskid/master

more detailed doc for embedding layer
starimpact · Apr 18, 2016 · 0e7d516 · 0e7d516
2 parents 2f2e80b + 76c978d
commit 0e7d516
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/operator/embedding.cc b/src/operator/embedding.cc
@@ -20,7 +20,9 @@ Operator* EmbeddingProp::CreateOperator(Context ctx) const {
 DMLC_REGISTER_PARAMETER(EmbeddingParam);
 
 MXNET_REGISTER_OP_PROPERTY(Embedding, EmbeddingProp)
-.describe("Get embedding for one-hot input")
+.describe("Get embedding for one-hot input. A n-dimensional input tensor will "
+"be trainsformed into a (n+1)-dimensional tensor, where a new dimension is "
+"added for the embedding results.")
 .add_argument("data", "Symbol", "Input data to the EmbeddingOp.")
 .add_argument("weight", "Symbol", "Enbedding weight matrix.")
 .add_arguments(EmbeddingParam::__FIELDS__());