diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc
index 4064190e19..4d791e19e3 100644
--- a/mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc
+++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/assign_cpu_kernel.cc
@@ -57,6 +57,10 @@ bool AssignCPUKernel::Launch(const std::vector<AddressPtr> &inputs, const std::v
   if (ret != 0) {
     MS_LOG(EXCEPTION) << "memcpy_s error, error no " << ret;
   }
+  ret = memcpy_s(outputs[0]->addr, max_size, inputs[1]->addr, total_size);
+  if (ret != 0) {
+    MS_LOG(EXCEPTION) << "memcpy_s error, error no " << ret;
+  }
   return true;
 }
 }  // namespace kernel