mindspore.ops.assign_add

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

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

variablevalue 遵循隐式类型转换规则,使数据类型一致。如果它们具有不同的数据类型,则低精度数据类型将转换为相对最高精度的数据类型。如果 value 是数值型,则会自动转换为Tensor,数据类型与计算中涉及的Tensor的数据类型保持一致。

说明

由于 variable 是数据类型为Parameter,数据类型不能更改,因此只允许 value 的数据类型转换为 variable 的数据类型。而且不同设备支持的转换类型会有所不同,建议在使用此函数时使用相同的数据类型。

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

  • value (Tensor) - 要和 variable 相加的值,shape必须与 variable 相同。建议在使用此函数时使用相同的数据类型。

返回:

Tensor,数据类型和shape与输入 variable 相同。

异常:
  • TypeError - 如果 value 既不是数值型也不是Tensor。

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

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor, ops
>>> from mindspore.common.initializer import initializer
>>> variable = mindspore.Parameter(initializer(1, [1], mindspore.int32), name="global_step")
>>> value = Tensor(np.ones([1]).astype(np.int32) * 100)
>>> ops.assign_add(variable, value)
>>> print(variable.asnumpy())
[101]