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.]]