mindspore.ops.assign

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

为网络参数赋值。

variable 和 value 遵循隐式类型转换规则，使数据类型一致。如果它们具有不同的数据类型，则低精度数据类型将转换为相对最高精度的数据类型。

参数：

  • variable (Parameter) - 网络参数。shape为 \((N,*)\) ，其中 \(*\) 表示任意数量的附加维度。

  • value (Tensor) - 要分配的值，shape与 variable 相同。

返回：

Tensor，数据类型和shape与 variable 相同。

异常：

  • TypeError - 如果 variable 不是Parameter。

  • TypeError - 如果 value 不是Tensor。

  • RuntimeError - 如果 variable 和 value 不支持参数的数据类型转换。

支持平台：

Ascend GPU CPU

样例：

>>> 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.]