mindspore.ops.Assign

class mindspore.ops.Assign(*args, **kwargs)[source]

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. \((N,*)\) where \(*\) means ,any number of additional dimensions, its rank should less than 8.

  • value (Tensor) - The value to be assigned, has the same shape with variable.

Outputs:

Tensor, has the same data type and shape as original variable.

Raises
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)
[2.]