fixed a bug in Paddle::Vector::createCpuVectorFromNumpy

8 years ago · 91e6dcb68f
parent 70fecee080
commit 91e6dcb68f
2 changed files with 2 additions and 1 deletions
--- 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);
  }
--- 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()
+