!3042 fix sens tensor check issue

Merge pull request !3042 from wangqiuliang/fix-sens-tensor-check-issue

mindspore/ccsrc/pynative/pynative_execute.cc

@@ -351,8 +351,8 @@ py::object RunOpInVM(const OpExecInfoPtr &op_exec_info, PynativeStatusCode *stat
     for (size_t i = 0; i < op_inputs.size(); i++) {
       py::object input = op_inputs[i];
       if (py::hasattr(input, "__parameter__")) {
-        result[i] = py::getattr(input, "data");
-      } else {
+        input = py::getattr(input, "data");
+      }
       auto tensor = py::cast<tensor::TensorPtr>(input);
       auto new_tensor = std::make_shared<tensor::Tensor>(tensor->data_type(), tensor->shape(), tensor->data_ptr());
       new_tensor->set_device_address(tensor->device_address());

mindspore/ops/composite/base.py

@@ -120,6 +120,9 @@ class GradOperation(GradOperation_):
         """ Pynative forward run to build grad graph. """
         if self.sens_param:
             args = args[:-1]
+        for arg in args:
+            if not isinstance(arg, Tensor):
+                raise TypeError("grad inputs should be tensor in pynative mode")
         if isinstance(fn, FunctionType):
             _pynative_exec.set_grad_flag(True)
             _pynative_exec.new_graph(fn, *args)
@@ -150,9 +153,6 @@ class GradOperation(GradOperation_):
             else:
                 @_wrap_func
                 def after_grad(*args):
-                    for arg in args:
-                        if not isinstance(arg, Tensor):
-                            raise TypeError("grad inputs should be tensor in pynative mode")
                     self._pynative_forward_run(args, fn)
                     _pynative_exec.grad(grad_, fn, weights, *args)
                     out = _pynative_exec(*args)