mindspore.ops.assign_add

查看源文件
mindspore.ops.assign_add(variable, value)[源代码]

通过加法运算更新网络参数或者tensor。

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

参数:
  • variable (Union[Parameter, Tensor]) - 输入的网络参数或者tensor。

  • value (Union[Tensor, Number]) - 要和 variable 相加的值。

返回:

Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> variable = mindspore.Parameter([1], name="global_step")
>>> value = mindspore.tensor([100], dtype=mindspore.int32)
>>> mindspore.ops.assign_add(variable, value)
>>> print(variable.asnumpy())
[101]