diff --git a/paddle/gserver/tests/test_Expand.cpp b/paddle/gserver/tests/test_Expand.cpp index a84a518a01e92081681ce508f8d680915b959d41..d32bf0152f77bba098daa508fe448784ac013549 100644 --- a/paddle/gserver/tests/test_Expand.cpp +++ b/paddle/gserver/tests/test_Expand.cpp @@ -91,6 +91,8 @@ TEST(Layer, ExpandLayerFwd) { doOneExpandTest("non-seq", false, useGpu, input1, input2, result); // CPU case 2. non-seq expand to sub-seq + // NOTE: input1.batch_size == input2.sequencelength in this case. + // i.e, input1 expands by input2.sequence // input1 = 1,2,3 // input2 = [[4,5]],[[6]],[[7],[8,9]] // result = [[1,1]],[[2]],[[3],[3,3]]