mindspore.ops.Assign

class mindspore.ops.Assign(*args, **kwargs)

Assigns Parameter with a value.

Inputs of variable and value comply with the implicit type conversion rules to make the data types consistent. If they have different data types, lower priority data type will be converted to relatively highest priority data type. RuntimeError exception will be thrown when the data type conversion of Parameter is required.

Inputs:

• variable (Parameter) - The Parameter.

• value (Tensor) - The value to be assigned.

Outputs:

Tensor, has the same type as original variable.

Raises

• TypeError – If variable is not a Parameter.

• TypeError – If value is not a Tensor.

Supported Platforms:

Ascend GPU CPU

Examples

>>> class Net(nn.Cell):
...     def __init__(self):
...         super(Net, self).__init__()
...         self.y = mindspore.Parameter(Tensor([1.0], mindspore.float32), name="y")
...
...     def construct(self, x):
...         ops.Assign()(self.y, x)
...         return self.y
...
>>> x = Tensor([2.0], mindspore.float32)
>>> net = Net()
>>> output = net(x)
>>> print(output)
Parameter (name=y, shape=(1,), dtype=Float32, requires_grad=True)