From 1c64b1193cab0ea0ca3e53ad978e089259b248cb Mon Sep 17 00:00:00 2001 From: Chiyuan Zhang Date: Mon, 18 Apr 2016 11:21:22 -0400 Subject: [PATCH] more detailed doc for embedding layer (#892) --- src/operator/embedding.cc | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/operator/embedding.cc b/src/operator/embedding.cc index 0906600f..c1b7f59e 100644 --- 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__());