From ccd21657c14e5d9cb3c8f8ef8cc1ce0fc621a4db Mon Sep 17 00:00:00 2001 From: Tink_Y <31891223+tink2123@users.noreply.github.com> Date: Thu, 11 Oct 2018 23:54:18 +0800 Subject: [PATCH] Update new_op_cn.md --- doc/fluid/dev/new_op_cn.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/doc/fluid/dev/new_op_cn.md b/doc/fluid/dev/new_op_cn.md index d0521342c..bbe296eda 100644 --- a/doc/fluid/dev/new_op_cn.md +++ b/doc/fluid/dev/new_op_cn.md @@ -184,8 +184,8 @@ MulOp(const std::string &type, const framework::VariableNameMap &inputs, 还需要重写`InferShape`接口。`InferShape`为const函数,不能修改Op的成员变量,参数为`const framework::InferShapeContext &ctx`,通过该参数可获取到输入输出以及属性。它的功能是: - - 1). 做检查, 尽早报错:检查输入数据维度、类型等是否合法。 - - 2). 设置输出Tensor的形状。 + - 做检查, 尽早报错:检查输入数据维度、类型等是否合法。 + - 设置输出Tensor的形状。 通常`OpProtoMaker`和`Op`类的定义写在`.cc`文件中,和下面将要介绍的注册函数一起放在`.cc`中 -- GitLab