From 6e5c44d3fe7a3e144f8fa396796a7219bf93e53e Mon Sep 17 00:00:00 2001 From: heqiaozhi Date: Wed, 10 Apr 2019 14:42:19 +0800 Subject: [PATCH] add doc --- paddle/fluid/operators/cvm_op.cc | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/operators/cvm_op.cc b/paddle/fluid/operators/cvm_op.cc index 0d66c10599..bbcb847ae6 100644 --- a/paddle/fluid/operators/cvm_op.cc +++ b/paddle/fluid/operators/cvm_op.cc @@ -114,8 +114,11 @@ class CVMOpMaker : public framework::OpProtoAndCheckerMaker { AddAttr("use_cvm", "bool, use cvm or not").SetDefault(true); AddComment(R"DOC( CVM Operator. + We assume that input is a embedding vector with cvm_feature(show and click), which shape is [N * D] (D is 2(cvm_feature) + embedding dim, N is batch_size) + if use_cvm is True, we will log(cvm_feature), and output shape is [N * D]. + if use_cvm is False, we will remove cvm_feature from input, and output shape is [N * (D - 2)]. - example: + Example: input = fluid.layers.data(name=\"input\", shape=[-1, 1], lod_level=1, append_batch_size=False, dtype=\"int64\") label = fluid.layers.data(name=\"label\", shape=[-1, 1], append_batch_size=False, dtype=\"int64\") -- GitLab