index_add_

paddle. index_add_ ( x, index, axis, value, name=None ) [源代码]

Inplace 版本的 index_add API,对输入 x 采用 Inplace 策略。

代码示例

>>> import paddle
>>> paddle.device.set_device('gpu')

>>> input_tensor = paddle.to_tensor(paddle.ones((3, 3)), dtype="float32")
>>> index = paddle.to_tensor([0, 2], dtype="int32")
>>> value = paddle.to_tensor([[1, 1], [1, 1], [1, 1]], dtype="float32")
>>> inplace_res = paddle.index_add_(input_tensor, index, 1, value)
>>> print(inplace_res)
Tensor(shape=[3, 3], dtype=float32, place=Place(gpu:0), stop_gradient=True,
[[2., 1., 2.],
 [2., 1., 2.],
 [2., 1., 2.]])

更多关于 inplace 操作的介绍请参考 3.1.3 原位(Inplace)操作和非原位操作的区别 了解详情。

使用本API的教程文档