diff --git a/paddle/api/Vector.cpp b/paddle/api/Vector.cpp index 3da7a5c476da26a1b8b64d8a35a871b7db7fe2e7..b8b1b2d2f1977b7f38c37d355ac9036034b95392 100644 --- a/paddle/api/Vector.cpp +++ b/paddle/api/Vector.cpp @@ -223,7 +223,7 @@ Vector* Vector::createCpuVectorFromNumpy(float* data, int dim, bool copy) { auto retVec = new Vector(); if (copy) { retVec->m->vec = paddle::Vector::create((size_t)dim, false); - return retVec; + retVec->m->vec->copyFrom(data, dim); } else { retVec->m->vec = paddle::Vector::create(data, (size_t)dim, false); } diff --git a/paddle/api/test/testVector.py b/paddle/api/test/testVector.py index 4903951414643de5d090f276cd0d5995f25238dc..5ca4d90dee7695dc11596efd5ee0c9b1ace14cfc 100644 --- a/paddle/api/test/testVector.py +++ b/paddle/api/test/testVector.py @@ -150,3 +150,4 @@ if __name__ == '__main__': swig_paddle.initPaddle("--use_gpu=1" if swig_paddle.isGpuVersion() else "--use_gpu=0") unittest.main() +