mindspore.ops.InplaceUpdateV2
- class mindspore.ops.InplaceUpdateV2[源代码]
根据 indices,将 x 中的某些值更新为 v。
警告
这是一个实验性API,后续可能修改或删除。
更多参考详见
mindspore.ops.inplace_update()
。- 输入:
x (Tensor) - 待更新的Tensor。数据类型支持float16、float32或int32。
indices (Union[int, tuple, Tensor]) - 指定将 x 的哪些行更新为 v 。可以为int或Tuple或Tensor,取值范围[0, len(x))。
v (Tensor) - 除第一个维度之外shape必须与 x 的shape相同。第一个维度必须与 indices 的长度相同。数据类型与 x 相同。
- 输出:
Tensor,与 x 的shape和数据类型相同。
- 支持平台:
GPU
CPU
样例:
>>> import numpy as np >>> import mindspore >>> from mindspore import Tensor, ops >>> indices = (0, 1) >>> x = Tensor(np.array([[1, 2], [3, 4], [5, 6]]), mindspore.float32) >>> v = Tensor(np.array([[0.5, 1.0], [1.0, 1.5]]), mindspore.float32) >>> inplace_update_v2 = ops.InplaceUpdateV2() >>> output = inplace_update_v2(x, indices, v) >>> print(output) [[0.5 1. ] [1. 1.5] [5. 6. ]]