mindspore.ops.assign

mindspore.ops.assign(variable, value)[源代码]

为网络参数或者tensor赋值。

支持隐式类型转换，类型提升。

参数：

variable (Union[Parameter, Tensor]) - 输入的网络参数或者tensor。
value (Tensor) - 要分配的值，shape与 variable 相同。

返回：

Tensor

支持平台：

Ascend GPU CPU

样例：

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