未验证 提交 a5167ce0 编写于 作者: K kexinzhao 提交者: GitHub

fix lod_array_lengh op equation (#6307)

上级 002a7b4d
......@@ -43,12 +43,16 @@ class LoDArrayLengthProtoMaker : public framework::OpProtoAndCheckerMaker {
: OpProtoAndCheckerMaker(proto, op_checker) {
AddInput("X", "(LoDTensorArray) The input tensor array.");
AddOutput("Out", "(Tensor) 1x1 CPU Tensor of length, int64_t");
AddComment(R"DOC(Get the length of lod tensor array
AddComment(R"DOC(
LoDArrayLength Operator.
Out = len(X)
This operator obtains the length of lod tensor array:
$$Out = len(X)$$
NOTE: The output is a CPU Tensor since the control variable should be only in
CPU and the length of LoDTensorArray should be used as control variables.
)DOC");
}
};
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册