fix lod_array_lengh op equation (#6307)

release/0.11.0
kexinzhao 8 years ago committed by GitHub
parent 002a7b4d01
commit a5167ce0b4
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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");
}
};

Loading…
Cancel
Save