mindspore.ops.assign

View Source On Gitee
mindspore.ops.assign(variable, value)[source]

Assigns Parameter or Tensor with a value.

Args of variable and value comply with the implicit type conversion rules to make the data types consistent. If they have different data types, the lower priority data type will be converted to the relatively highest priority data type.

Parameters

  • variable (Union[Parameter, Tensor]) – The Parameter or Tensor. \((N,*)\) where \(*\) means, any number of additional dimensions.

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

Returns

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

Raises

  • TypeError – If variable is neither a Parameter nor a Tensor.

  • TypeError – If value is not a Tensor.

  • RuntimeError – If the data type of variable and value conversion of Parameter or Tensor is required when data type conversion of Parameter is not supported.

Supported Platforms:

Ascend GPU CPU

Examples

>>> import mindspore
>>> from mindspore import Tensor, ops
>>> value = Tensor([2.0], mindspore.float32)
>>> variable = mindspore.Parameter(Tensor([1.0], mindspore.float32), name="variable")
>>> ops.assign(variable, value)
>>> print(variable.asnumpy())
[2.]