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]