mindspore.Tensor.itemset

mindspore.Tensor.itemset(*args)

将标量插入到Tensor（并将标量转换为Tensor的数据类型）。

至少有1个参数，并且最后一个参数被定义为设定值。 Tensor.itemset(*args)等同于 \(Tensor[args] = item\) 。

参数：

• args (Union[(numbers.Number), (int/tuple(int), numbers.Number)]) - 指定索引和值的参数。如果 args 包含一个参数（标量），则其仅在Tensor大小为1的情况下使用。如果 args 包含两个参数，则最后一个参数是要设置的值且必须是标量，而第一个参数指定单个Tensor元素的位置。参数值是整数或者元组。

返回：

一个新的Tensor，其值为 \(Tensor[args] = item\) 。

异常：

• ValueError - 第一个参数的长度不等于Tensor的ndim。

• IndexError - 只提供了一个参数，并且原来的Tensor不是标量。

支持平台：

Ascend GPU CPU

样例：

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1,2,3],[4,5,6]], dtype=np.float32))
>>> print(x.itemset((0,1), 4))
[[1. 4. 3.]
[4. 5. 6.]]
>>> print(x)
[[1. 2. 3.]
[4. 5. 6.]]