add support for parameter

support for tensor setitem

add support for tensor assgin

mindspore/ccsrc/ir/tensor.cc

@@ -92,6 +92,10 @@ Tensor &Tensor::operator=(const Tensor &tensor) {
   }
   return *this;
 }
+Tensor &Tensor::AssignValue(const Tensor &tensor) {
+  *this = tensor;
+  return *this;
+}
 
 bool Tensor::operator==(const Tensor &tensor) const {
   return (MetaTensor::operator==(tensor) && data_ == tensor.data_);
@@ -470,6 +474,19 @@ REGISTER_PYBIND_DEFINE(Tensor, ([](const py::module *m) {
                                  >>> data.set_dtype(mindspore.int32)
                                  mindspore.int32
                              )mydelimiter")
+                           .def("assign_value", &Tensor::AssignValue, R"mydelimiter(
+                             Assign another tensor value to this.
+
+                             Arg:
+                                 value (:class:`mindspore.tensor`): The value tensor.
+
+                             Examples:
+                                 >>> data = mindspore.Tensor(np.ones((1, 2), np.float32))
+                                 >>> data2 = mindspore.Tensor(np.ones((2, 2), np.float32))
+                                 >>> data.assign_value(data2)
+                                 >>> data.shape
+                                 (2, 2)
+                             )mydelimiter")
                            .def("__str__", &Tensor::ToString)
                            .def("__repr__", &Tensor::ToStringRepr)
                            .def(py::pickle(

mindspore/ccsrc/ir/tensor.h

@@ -173,6 +173,9 @@ class Tensor : public MetaTensor {
   // It is different from 'operator==' which just compare shape/type/address, it do real value comparison.
   bool ValueEqual(const Tensor &other) const;
 
+  // assgin value to this tensor
+  Tensor &AssignValue(const Tensor &tensor);
+
   bool operator==(const Value &other) const override {
     if (other.isa<Tensor>()) {
       auto other_ = static_cast<const Tensor &>(other);

mindspore/common/parameter.py

@@ -203,6 +203,8 @@ class Parameter:
         return self.default_input / other
 
     def __setitem__(self, index, value):
+        default_input = self.default_input
+        default_input[index] = value
         return self
 
     def set_parameter_data(self, data):

mindspore/common/tensor.py

@@ -150,6 +150,8 @@ class Tensor(Tensor_):
         return out
 
     def __setitem__(self, index, value):
+        out = tensor_operator_registry.get('__setitem__')(self, index, value)
+        self.assign_value(out)
         return self
 
     def __gt__(self, other):