mindspore.Tensor

class mindspore.Tensor(input_data=None, dtype=None, shape=None, init=None, internal=False)[源代码]

张量,即存储多维数组(n-dimensional array)的数据结构。

参数:

  • input_data (Union[Tensor, float, int, bool, tuple, list, numpy.ndarray]) - 被存储的数据,可以是其它Tensor,也可以是Python基本数据(如int,float,bool等),或是一个NumPy对象。默认值:None。

  • dtype (mindspore.dtype) - 用于定义该Tensor的数据类型,必须是 mindspore.dtype 中定义的类型。如果该参数为None,则数据类型与 input_data 一致,默认值:None。

  • shape (Union[tuple, list, int]) - 用于定义该Tensor的形状。如果指定了 input_data ,则无需设置该参数。默认值:None。

  • init (Initializer) - 用于在并行模式中延迟Tensor的数据的初始化,如果指定该参数,则 dtypeshape 也必须被指定。不推荐在非自动并行之外的场景下使用该接口。只有当调用 Tensor.init_data 时,才会使用指定的 init 来初始化Tensor数据。默认值:None。

  • internal (bool) - Tensor是否由框架创建。如果为True,表示Tensor是由框架创建的,如果为False,表示Tensor是由用户创建的。默认值:False。

输出:

Tensor。

样例:

>>> import numpy as np
>>> import mindspore as ms
>>> from mindspore import Tensor
>>> from mindspore.common.initializer import One
>>> # initialize a tensor with numpy.ndarray
>>> t1 = Tensor(np.zeros([1, 2, 3]), ms.float32)
>>> print(t1)
[[[0. 0. 0.]
[0. 0. 0.]]]
>>> print(type(t1))
<class 'mindspore.common.tensor.Tensor'>
>>> print(t1.shape)
(1, 2, 3)
>>> print(t1.dtype)
Float32
>>>
>>> # initialize a tensor with a float scalar
>>> t2 = Tensor(0.1)
>>> print(t2)
0.1
>>> print(type(t2))
<class 'mindspore.common.tensor.Tensor'>
>>> print(t2.shape)
()
>>> print(t2.dtype)
Float32
>>>
>>> # initialize a tensor with a tuple
>>> t3 = Tensor((1, 2))
>>> print(t3)
[1 2]
>>> print(type(t3))
<class 'mindspore.common.tensor.Tensor'>
>>> print(t3.shape)
(2,)
>>> print(t3.dtype)
Int64
...
>>> # initialize a tensor with init
>>> t4 = Tensor(shape = (1, 3), dtype=ms.float32, init=One())
>>> print(t4)
[[1. 1. 1.]]
>>> print(type(t4))
<class 'mindspore.common.tensor.Tensor'>
>>> print(t4.shape)
(1, 3)
>>> print(t4.dtype)
Float32
property T

返回转置后的Tensor。

abs()[源代码]

返回每个元素的绝对值。

返回:

Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> a = Tensor([1.1, -2.1]).astype("float32")
>>> output = a.abs()
>>> print(output)
[1.1 2.1]
all(axis=(), keep_dims=False)[源代码]

检查在指定轴上所有元素是否均为True。

参数:

  • axis (Union[None, int, tuple(int)]) - 计算all的维度。当 axis 为None或者空元组的时候,计算所有维度。当 axis 为int或tuple(int)时,记Tensor的维度为dim,则其取值范围为[-dim, dim)。默认值:()。

  • keep_dims (bool) - 计算结果是否保留维度。默认值:False。

返回:

Tensor。如果在指定轴方向上所有数组元素都为True,则其值为True,否则其值为False。如果轴为None或空元组,则默认降维。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> a = Tensor([True, True, False])
>>> output = a.all()
>>> print(output)
False
any(axis=(), keep_dims=False)[源代码]

检查在指定轴方向上是否存在任意为True的Tensor元素。

参数:

  • axis (Union[None, int, tuple(int)]) - 计算any的维度。当 axis 为None或空元组时,计算所有维度。当 axis 为int或tuple(int)时,记Tensor的维度为dim,则其取值范围为[-dim, dim)。默认值:()。

  • keep_dims (bool) - 计算结果是否保留维度。默认值:False。

返回:

Tensor。如果在指定轴方向上所有Tensor元素都为True,则其值为True,否则其值为False。如果轴为None或空元组,则默认降维。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> a = Tensor([True, True, False])
>>> output = a.any()
>>> print(output)
True
col2im(output_size, kernel_size, dilation, padding_value, stride)[源代码]

将一组滑动的局部块组合成一个大张量。

参数:

  • output_size (Tensor) - 输出张量的后两维的shape。

  • kernel_size (Union[int, tuple[int], list[int]]) - 滑动窗口的大小。

  • dilation (Union[int, tuple[int], list[int]]) - 滑动窗口扩张的大小。

  • padding_value (Union[int, tuple[int], list[int]]) - 填充的大小。

  • stride (Union[int, tuple[int], list[int]]) - 步长的大小。

返回:

Tensor,输出的张量,维度和类型和输入一致。

异常:

  • TypeError - 如果 kernel_sizedilationpadding_valuestride 不属于 Union[int, tuple[int], list[int]]。

  • ValueError - 如果 kernel_sizedilationstride 值小于等于0或者个数大于2。

  • ValueError - 如果 padding_value 值小于0或者个数大于2。

支持平台:

GPU

样例:

>>> x = Tensor(input_data=np.random.rand(16, 16, 4, 25), dtype=mstype.float32)
>>> output_size = Tensor(input_data=[8, 8], dtype=mstype.int32)
>>> y = x.col2im(output_size, kernel_size=[2, 2], dilation=[2, 2], padding_value=[2, 2], stride=[2, 2])
>>> print(y.shape)
(16, 16, 8, 8)
argmax(axis=None)[源代码]

返回指定轴上最大值的索引。

参数:

  • axis (int, optional) - 默认情况下,返回扁平化Tensor的最大值序号,否则返回指定轴方向上。

返回:

Tensor,最大值的索引。它与原始Tensor具有相同的shape,但移除了轴方向上的维度。

异常:

  • ValueError - 入参axis的设定值超出了范围。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> a = Tensor(np.arange(10, 16).reshape(2, 3).astype("float32"))
>>> print(a.argmax())
5
argmin(axis=None)[源代码]

返回指定轴上最小值的索引。

参数:

  • axis (int, optional) - 返回扁平化Tensor的最小值序号,否则返回指定轴方向上的最小值序号。默认值: None。

返回:

Tensor,最小Tensor的索引。它与原始Tensor具有相同的shape,但移除了轴方向上的维度。

异常:

  • ValueError - 入参axis的设定值超出了范围。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> a = Tensor(np.arange(10, 16).reshape(2, 3).astype("float32"))
>>> print(a.argmin())
0
asnumpy()[源代码]

将张量转换为NumPy数组。该方法会将Tensor本身转换为NumPy的ndarray。这个Tensor和函数返回的ndarray共享内存地址。对Tensor本身的修改会反映到相应的ndarray上。

返回:

NumPy的ndarray,该ndarray与Tensor共享内存地址。

样例:

>>> from mindspore import Tensor
>>> import numpy as np
>>> x = Tensor(np.array([1, 2], dtype=np.float32))
>>> y = x.asnumpy()
>>> y[0] = 11
>>> print(x)
[11.  2.]
>>> print(y)
[11.  2.]
assign_value(value)[源代码]

将另一个Tensor的值赋给当前Tensor。

参数:

  • value (Tensor) - 用于赋值的Tensor。

返回:

Tensor,赋值后的Tensor。

astype(dtype, copy=True)[源代码]

将Tensor转为指定数据类型,可指定是否返回副本。

参数:

  • dtype (Union[mindspore.dtype , numpy.dtype , str]) - 指定的Tensor数据类型,可以是: mindspore.dtype.float32 , numpy.float32float32 的格式。默认值:mindspore.dtype.float32

  • copy (bool, optional) - 默认情况下,astype返回新拷贝的Tensor。如果该参数设为False,则返回输入Tensor而不是副本。默认值:True。

返回:

Tensor,指定数据类型的Tensor。

异常:

  • TypeError - 指定了无法解析的类型。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((1,2,2,1), dtype=np.float32))
>>> x = x.astype("int32")
>>> print(x.dtype)
Int32
atan2(y)[源代码]

逐元素计算x/y的反正切值。

x 指的当前 Tensor。

返回 \(\theta\ \in\ [-\pi, \pi]\) ,使得 \(x = r*\sin(\theta), y = r*\cos(\theta)\) ,其中 \(r = \sqrt{x^2 + y^2}\) 。 输入 xy 会通过隐式数据类型转换使数据类型保持一致。如果数据类型不同,低精度的数据类型会被转换到高精度的数据类型。

参数:

  • y (Tensor) - 输入Tensor。shape应能在广播后与 x 相同,或 x 的shape在广播后与 y 相同。

返回:

Tensor,与广播后的输入shape相同,和 x 数据类型相同。

异常:

  • TypeError - xy 不是Tensor。

  • RuntimeError - xy 之间的数据类型转换不被支持。

支持平台:

Ascend CPU GPU

样例:

>>> x = Tensor(np.array([0, 1]), mindspore.float32)
>>> y = Tensor(np.array([1, 1]), mindspore.float32)
>>> output = x.atan2(y)
>>> print(output)
[0.        0.7853982]
bernoulli(p=0.5, seed=- 1)[源代码]

以p的概率随机将输出的元素设置为0或1,服从伯努利分布。

\[out_{i} \sim Bernoulli(p_{i})\]

参数:

  • p (Union[Tensor, float], 可选) - shape需要可以被广播到当前Tensor。其数据类型为float32或float64。p 中每个值代表输出Tensor中对应广播位置为1的概率,数值范围在0到1之间。默认值:0.5。

  • seed (int, 可选) - 随机种子,用于生成随机数,数值范围是-1或正整数。默认值:-1,代表取当前时间戳。

返回:

Tensor,shape和数据类型与当前Tensor相同。

异常:

  • TypeError - 当前Tensor的数据类型不在int8, uint8, int16, int32, int64, bool, float32和float64中。

  • TypeError - p 的数据类型既不是float32也不是float64。

  • TypeError - seed 不是int。

  • ValueError - seed 是负数且不为-1。

  • ValueError - p 数值范围不在0到1之间。

支持平台:

GPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([1, 2, 3]), mindspore.int8)
>>> output = input_x.bernoulli(p=1.0)
>>> print(output)
[1 1 1]
>>> input_p = Tensor(np.array([0.0, 1.0, 1.0]), mindspore.float32)
>>> output = input_x.bernoulli(input_p)
>>> print(output)
[0 1 1]
bitwise_and(x)[源代码]

逐元素执行两个Tensor的与运算。

更多细节参考 mindspore.ops.bitwise_and()

参数:

  • x (Tensor) - 输入Tensor,是一个数据类型为uint16、int16或int32的Tensor。

返回:

Tensor,是一个与 x 相同类型的Tensor。

支持平台:

Ascend CPU

样例:

>>> from mindspore import Tensor
>>> import numpy as np
>>> a = Tensor(np.array([0, 0, 1, -1, 1, 1, 1]), mindspore.int16)
>>> b = Tensor(np.array([0, 1, 1, -1, -1, 2, 3]), mindspore.int16)
>>> output = a.bitwise_and(b)
>>> print(output)
[ 0  0  1 -1  1  0  1]
bitwise_or(x)[源代码]

逐元素执行两个Tensor的或运算。

更多细节参考 mindspore.ops.bitwise_or()

参数:

  • x (Tensor) - 输入Tensor,是一个数据类型为uint16、int16或int32的Tensor。

返回:

Tensor,是一个与 x 相同类型的Tensor。

支持平台:

Ascend CPU

样例:

>>> from mindspore import Tensor
>>> import numpy as np
>>> a = Tensor(np.array([0, 0, 1, -1, 1, 1, 1]), mindspore.int16)
>>> b = Tensor(np.array([0, 1, 1, -1, -1, 2, 3]), mindspore.int16)
>>> output = a.bitwise_or(b)
>>> print(output)
[ 0  1  1 -1 -1  3  3]
bitwise_xor(x)[源代码]

逐元素执行两个Tensor的异或运算。

更多细节参考 mindspore.ops.bitwise_xor()

参数:

  • x (Tensor) - 输入Tensor,是一个数据类型为uint16、int16或int32的Tensor。

返回:

Tensor,是一个与 x 相同类型的Tensor。

支持平台:

Ascend CPU

样例:

>>> from mindspore import Tensor
>>> import numpy as np
>>> a = Tensor(np.array([0, 0, 1, -1, 1, 1, 1]), mindspore.int16)
>>> b = Tensor(np.array([0, 1, 1, -1, -1, 2, 3]), mindspore.int16)
>>> output = a.bitwise_xor(b)
>>> print(output)
[ 0  1  0  0 -2  3  2]
broadcast_to(shape)[源代码]

将输入shape广播到目标shape。

更多细节请参考 mindspore.ops.broadcast_to()

参数:

  • shape (tuple) - 要广播的目标形状。可以由用户指定,或在要广播的维度上指定-1,它将被该位置的输入张量形状替换。

返回:

Tensor,形状为用户指定的 shape,类型和 self 相同。

异常:

  • TypeError - 如果输入的 shape 参数不是tuple类型。

  • ValueError - 如果输入的 shapeself 的形状不兼容,或者目标 shape 中的-1位于无效位置。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> from mindspore import dtype as mstype
>>> shape = (2, 3)
>>> x = Tensor(np.array([1, 2, 3]).astype(np.float32))
>>> output = x.broadcast_to(shape)
>>> print(output)
[[1. 2. 3.]
 [1. 2. 3.]]
>>> shape = (-1, 2)
>>> x = Tensor(np.array([[1], [2]]).astype(np.float32))
>>> output = x.broadcast_to(shape)
>>> print(output)
[[1. 1.]
 [2. 2.]]
ceil()[源代码]

向上取整。

返回:

Tensor。向上取整的结果。

异常:

  • TypeError - 如果当前Tensor的数据类型不是float16或者float32。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> a = Tensor([1.1, 2.5, -1.5]).astype("float32")
>>> output = a.ceil()
>>> print(output)
[ 2.  3. -1.]
choose(choices, mode='clip')[源代码]

根据原始Tensor数组和一个索引数组构造一个新的Tensor。

参数:

  • choices (Union[tuple, list, Tensor]) - 索引选择数组。原始输入Tensor和 choices 的广播维度必须相同。如果 choices 本身是一个Tensor,则其最外层的维度(即,对应于第0维的维度)被用来定义 choices 数组。

  • mode (‘raise’, ‘wrap’, ‘clip’, optional) - 指定如何处理 [0, n-1] 外部的索引:

    • raise - 引发异常(默认);

    • wrap - 原值映射为对n取余后的值;

    • clip - 大于n-1的值会被映射为n-1。该模式下禁用负数索引。

返回:

Tensor,合并后的结果。

异常:

  • ValueError - 输入Tensor和任一 choices 无法广播。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> choices = [[0, 1, 2, 3], [10, 11, 12, 13], [20, 21, 22, 23], [30, 31, 32, 33]]
>>> x = Tensor(np.array([2, 3, 1, 0]))
>>> print(x.choose(choices))
[20 31 12  3]
clip(xmin, xmax, dtype=None)[源代码]

裁剪Tensor中的值。

给定一个区间,区间外的值将被裁剪到区间边缘。 例如,如果指定的间隔为 \([0, 1]\) ,则小于0的值将变为0,大于1的值将变为1。

Note

目前不支持裁剪 xmin=nanxmax=nan

参数:

  • xmin (Tensor, scalar, None) - 最小值。如果值为None,则不在间隔的下边缘执行裁剪操作。xminxmax 只能有一个为None。

  • xmax (Tensor, scalar, None) - 最大值。如果值为None,则不在间隔的上边缘执行裁剪操作。xminxmax 只能有一个为None。如果 xminxmax 是Tensor,则三个Tensor将被广播进行shape匹配。

  • dtype (mindspore.dtype , optional) - 覆盖输出Tensor的dtype。默认值为None。

返回:

Tensor,含有输入Tensor的元素,其中values < xmin 被替换为 xmin ,values > xmax 被替换为 xmax

异常:

  • TypeError - 输入的类型与Tensor不一致。

  • ValueError - 输入与Tensor的shape不能广播,或者 xminxmax 都是 None

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> x = Tensor([1, 2, 3, -4, 0, 3, 2, 0]).astype("float32")
>>> y = x.clip(0, 2)
>>> print(y)
[1. 2. 2. 0. 0. 2. 2. 0.]
>>> t = Tensor([1, 1, 1, 1, 1, 1, 1, 1])
>>> y = x.clip(t, 2)
>>> print(y)
[1. 2. 2. 1. 1. 2. 2. 1.]
copy()[源代码]

复制一个Tensor并返回。

Note

当前实现不支持类似NumPy的 order 参数。

返回:

复制的Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> a = Tensor(np.ones((3,3)).astype("float32"))
>>> output = a.copy()
>>> print(output)
[[1. 1. 1.]
[1. 1. 1.]
[1. 1. 1.]]
cosh()[源代码]

逐元素计算双曲余弦值。

\[out_i = cosh(x_i)\]

返回:

Tensor,数据类型和shape与 x 相同。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> a = Tensor(np.array([0.24, 0.83, 0.31, 0.09]), mindspore.float32)
>>> output = a.cosh()
>>> print(output)
[1.0289385 1.364684 1.048436 1.0040528]
cummax(axis)[源代码]

返回一个元组(最值、索引),其中最值是输入张量 x 沿维度 axis 的累积最大值,索引是每个最大值的索引位置。

\[\begin{split}\begin{array}{ll} \\ y{i} = max(x{1}, x{2}, ... , x{i}) \end{array}\end{split}\]

参数:

  • axis (int) - 算子操作的维度,维度的大小范围是[-x.ndim, x.ndim - 1]。

返回:

一个包含两个Tensor的元组,分别表示累积最大值和对应索引。

异常:

  • TypeError - 如果 axis 不是int。

  • ValueError - 如果 axis 不在范围[-x.ndim, x.ndim - 1]内。

支持平台:

GPU CPU

样例:

>>> import mindspore
>>> import numpy as np
>>> from mindspore import Tensor
>>> import mindspore.ops as ops
>>> x = Tensor(np.array([[3, 4, 6, 10], [1, 6, 7, 9], [4, 3, 8, 7], [1, 3, 7, 9]]).astype(np.float32))
>>> output = x.cummax(axis=0)
>>> print(output[0])
[[ 3.  4.  6. 10.]
 [ 3.  6.  7. 10.]
 [ 4.  6.  8. 10.]
 [ 4.  6.  8. 10.]]
>>> print(output[1])
[[0 0 0 0]
 [0 1 1 0]
 [2 1 2 0]
 [2 1 2 0]]
cummin(axis)[源代码]

返回一个元组(最值、索引),其中最值是输入张量 x 沿维度 axis 的累积最小值,索引是每个最小值的索引位置。

\[\begin{split}\begin{array}{ll} \\ y{i} = min(x{1}, x{2}, ... , x{i}) \end{array}\end{split}\]

参数:

  • axis (int) - 算子操作的维度,维度的大小范围是[-x.ndim, x.ndim - 1]。

返回:

一个包含两个Tensor的元组,分别表示累积最小值和对应索引。

异常:

  • TypeError - 如果 axis 不是int。

  • ValueError - 如果 axis 不在范围[-x.ndim, x.ndim - 1]内。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor, ops
>>> import mindspore
>>> a = Tensor([-0.2284, -0.6628,  0.0975,  0.2680, -1.3298, -0.4220], mindspore.float32)
>>> output = a.cummin(axis=0)
>>> print(output[0])
[-0.2284 -0.6628 -0.6628 -0.6628 -1.3298 -1.3298]
>>> print(output[1])
[0 1 1 1 4 4]
cumsum(axis=None, dtype=None)[源代码]

返回指定轴方向上元素的累加值。

Note

如果 dtypeint8 , int16bool ,则结果 dtype 将提升为 int32 ,不支持 int64

参数:

  • axis (int, optional) - 轴,在该轴方向上的累积和。默认情况下,计算所有元素的累加和。

  • dtype (mindspore.dtype , optional) - 如果未指定参数值,则保持与原始Tensor相同,除非参数值是一个精度小于 float32 的整数。在这种情况下,使用 float32 。默认值:None。

异常:

  • ValueError - 轴超出范围。

返回:

Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> a = Tensor(np.ones((3,3)).astype("float32"))
>>> output = a.cumsum(axis=0)
>>> print(output)
[[1. 1. 1.]
[2. 2. 2.]
[3. 3. 3.]]
diag()[源代码]

用给定的对角线值构造对角线张量。

假设输入Tensor维度为 \([D_1,... D_k]\) ,则输出是一个rank为2k的tensor,其维度为 \([D_1,..., D_k, D_1,..., D_k]\) ,其中 \(output[i_1,..., i_k, i_1,..., i_k] = self[i_1,..., i_k]\) 并且其他位置的值为0。

返回:

Tensor,具有与输入Tensor相同的数据类型。

异常:

  • ValueError - 输入Tensor的rank小于1。

支持平台:

Ascend GPU

样例:

>>> from mindspore import Tensor
>>> x = Tensor([1, 2, 3, 4]).astype('int32')
>>> output = x.diag()
>>> print(output)
[[1, 0, 0, 0]
 [0, 2, 0, 0]
 [0, 0, 3, 0]
 [0, 0, 0, 4]]
diagonal(offset=0, axis1=0, axis2=1)[源代码]

返回指定的对角线。

参数:

  • offset (int, optional) - 对角线与主对角线的偏移。可以是正值或负值。默认为主对角线。

  • axis1 (int, optional) - 二维子数组的第一轴,对角线应该从这里开始。默认为第一轴(0)。

  • axis2 (int, optional) - 二维子数组的第二轴,对角线应该从这里开始。默认为第二轴。

返回:

Tensor,如果Tensor是二维,则返回值是一维数组。

异常:

  • ValueError - 输入Tensor的维度少于2。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> a = Tensor(np.arange(4).reshape(2, 2))
>>> print(a)
[[0 1]
[2 3]]
>>> output = a.diagonal()
>>> print(output)
[0 3]
property dtype

返回张量的数据类型(mindspore.dtype)。

expand_as(x)[源代码]

将目标张量的维度扩展为输入张量的维度。

参数:

  • x (Tensor) - 输入的张量。

返回:

维度与输入张量的相同的Tensor。输出张量的维度必须遵守广播规则。广播规则指输出张量的维度需要扩展为输入张量的维度,如果目标张量的维度大于输入张量的维度,则不满足广播规则。

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore import dtype as mstype
>>> x = Tensor([1, 2, 3], dtype=mstype.float32)
>>> y = Tensor(np.ones((2, 3)), dtype=mstype.float32)
>>> output = x.expand_as(y)
>>> print(output)
[[1. 2. 3.]
[1. 2. 3.]]
expand_dims(axis)[源代码]

沿指定轴扩展Tensor维度。

参数:

  • axis (int) - 扩展维度指定的轴。

返回:

Tensor,指定轴上扩展的维度为1。

异常:

  • TypeError - axis不是int类型。

  • ValueError - axis的取值不在[-self.ndim - 1, self.ndim + 1)范围内。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((2,2), dtype=np.float32))
>>> print(x)
[[1. 1.]
[1. 1.]]
>>> print(x.shape)
(2, 2)
>>> y = x.expand_dims(axis=0)
>>> print(y)
[[[1. 1.]
[1. 1.]]]
>>> print(y.shape)
(1, 2, 2)
erf()[源代码]

逐元素计算原Tensor的高斯误差函数。 更多细节参考 mindspore.ops.erf()

返回:

Tensor,具有与原Tensor相同的数据类型和shape。

异常:

  • TypeError - 原Tensor的数据类型既不是float16也不是float32。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([-1, 0, 1, 2, 3]), mindspore.float32)
>>> output = x.erf()
>>> print(output)
[-0.8427168   0.          0.8427168   0.99530876  0.99997765]
erfc()[源代码]

逐元素计算原Tensor的互补误差函数。 更多细节参考 mindspore.ops.erfc()

返回:

Tensor,具有与原Tensor相同的数据类型和shape。

异常:

  • TypeError - 原Tensor的数据类型既不是float16也不是float32。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([-1, 0, 1, 2, 3]), mindspore.float32)
>>> output = x.erfc()
>>> print(output)
[1.8427168e+00 1.0000000e+00 1.5728319e-01 4.6912432e-03 2.2351742e-05]
fill(value)[源代码]

用标量值填充数组。

Note

与NumPy不同,Tensor.fill()将始终返回一个新的Tensor,而不是填充原来的Tensor。

参数:

  • value (Union[None, int, float, bool]) - 所有元素都被赋予这个值。

返回:

Tensor,与原来的dtype和shape相同的Tensor。

异常:

  • TypeError - 输入参数具有前面未指定的类型。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> a = Tensor(np.arange(4).reshape((2,2)).astype('float32'))
>>> print(a.fill(1.0))
[[1. 1.]
[1. 1.]]
flatten(order='C')[源代码]

返回展开成一维的Tensor的副本。

参数:

order (str, optional) - 可以在’C’和’F’之间进行选择。’C’表示按行优先(C风格)顺序展开。’F’表示按列优先顺序(Fortran风格)进行扁平化。仅支持’C’和’F’。默认值:’C’。

返回:

Tensor,具有与输入相同的数据类型。

异常:

  • TypeError - order 不是字符串类型。

  • ValueError - order 是字符串类型,但不是’C’或’F’。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((2,3,4), dtype=np.float32))
>>> output = x.flatten()
>>> print(output.shape)
(24,)
flush_from_cache()[源代码]

如果Tensor开启缓存作用,则将缓存数据刷新到host侧。

样例:

>>> from mindspore import Tensor
>>> import numpy as np
>>> x = Tensor(np.array([1, 2], dtype=np.float32))
>>> y = x.flush_from_cache()
>>> print(y)
None
static from_numpy(array)[源代码]

通过不复制数据的方式将Numpy数组转换为张量。

参数:

array (numpy.array) - 输入数组。

返回:

与输入的张量具有相同的数据类型的Tensor。

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = np.array([1, 2])
>>> output = Tensor.from_numpy(x)
>>> print(output)
[1 2]
gather_elements(dim, index)[源代码]

获取指定轴的元素。

对于三维Tensor,输出为:

output[i][j][k] = x[index[i][j][k]][j][k]  # if dim == 0

output[i][j][k] = x[i][index[i][j][k]][k]  # if dim == 1

output[i][j][k] = x[i][j][index[i][j][k]]  # if dim == 2

index 与当前Tensor拥有一样的维度长度,且除 dim 维外其他维度一致。如果维度 dim 为i,当前Tensor是shape为 \((z_0, z_1, ..., z_i, ..., z_{n-1})\) 的n维Tensor,则 index 必须是shape为 \((z_0, z_1, ..., y, ..., z_{n-1})\) 的n维Tensor,其中 y 大于等于1。输出的shape与 index 相同。

参数:

  • dim (int) - 获取元素的轴。数据类型为int32或int64。取值范围为[-self.ndim, self.ndim)。

  • index (Tensor) - 获取收集元素的索引。支持的数据类型包括:int32,int64。每个索引元素的取值范围为[-self.shape(dim), self.shape(dim))。

返回:

Tensor,shape与 index 相同,即其shape为 \((z_0, z_1, ..., y, ..., z_{n-1})\),数据类型与 self.dtype 相同。

异常:

  • TypeError - dimindex 的数据类型既不是int32,也不是int64。

  • ValueError - selfindex 的维度长度不一致。

  • ValueError - selfindexdim 维外的维度不一致。

  • ValueError - dim 的值不在合理范围内。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> import mindspore
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1, 2], [3, 4]]), mindspore.int32)
>>> index = Tensor(np.array([[0, 0], [1, 0]]), mindspore.int32)
>>> dim = 1
>>> output = x.gather_elements(dim, index)
>>> print(output)
[[1 1]
 [4 3]]
gather_nd(indices)[源代码]

按索引从输入Tensor中获取切片。 使用给定的索引从具有指定形状的输入Tensor中搜集切片。 输入Tensor的shape是 \((N,*)\) ,其中 \(*\) 表示任意数量的附加维度。下文中的 input_x 代指输入Tensor本身。 indices 是一个K维的整数张量,假定它的K-1维张量中的每一个元素是输入Tensor的切片,那么有:

\[output[(i_0, ..., i_{K-2})] = input\_x[indices[(i_0, ..., i_{K-2})]]\]

indices 的最后一维不能超过输入Tensor的秩: \(indices.shape[-1] <= input\_x.rank\)

参数:

  • indices (Tensor) - 获取收集元素的索引张量,其数据类型包括:int32,int64。

返回:

Tensor,具有与输入Tensor相同的数据类型,shape维度为 \(indices\_shape[:-1] + input\_x\_shape[indices\_shape[-1]:]\)

异常:

  • ValueError - 如果输入Tensor的shape长度小于 indices 的最后一个维度。

支持平台:

Ascend GPU CPU

样例:

>>> input_x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]), mindspore.float32)
>>> indices = Tensor(np.array([[0, 0], [1, 1]]), mindspore.int32)
>>> output = input_x.gather_nd(indices)
>>> print(output)
[-0.1  0.5]
gather(input_indices, axis)[源代码]

返回指定 axisinput_indices 的元素对应的输入Tensor切片。为了方便描述,对于输入Tensor记为 input_params

Note

  1. input_indices 的值必须在 [0, input_params.shape[axis]) 的范围内,结果未定义超出范围。

  2. 当前在Ascend平台,input_params的值不能是 bool_ 类型。

参数:

  • input_indices (Tensor) - 待切片的索引张量,其形状为 \((y_1, y_2, ..., y_S)\),代表指定原始张量元素的索引,其数据类型包括:int32,int64。

  • axis (int) - 指定维度索引的轴以搜集切片。

返回:

Tensor,其中shape维度为 \(input\_params.shape[:axis] + input\_indices.shape + input\_params.shape[axis + 1:]\)

异常:

  • TypeError - 如果 axis 不是一个整数。

  • TypeError - 如果 input_indices 不是一个整数类型的Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> # case1: input_indices is a Tensor with shape (5, ).
>>> input_params = Tensor(np.array([1, 2, 3, 4, 5, 6, 7]), mindspore.float32)
>>> input_indices = Tensor(np.array([0, 2, 4, 2, 6]), mindspore.int32)
>>> axis = 0
>>> output = input_params.gather(input_indices, axis)
>>> print(output)
[1. 3. 5. 3. 7.]
>>> # case2: input_indices is a Tensor with shape (2, 2). When the input_params has one dimension,
>>> # the output shape is equal to the input_indices shape.
>>> input_indices = Tensor(np.array([[0, 2], [2, 6]]), mindspore.int32)
>>> axis = 0
>>> output = input_params.gather(input_indices, axis)
>>> print(output)
[[ 1. 3.]
 [ 3. 7.]]
>>> # case3: input_indices is a Tensor with shape (2, ) and
>>> # input_params is a Tensor with shape (3, 4) and axis is 0.
>>> input_params = Tensor(np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]), mindspore.float32)
>>> input_indices = Tensor(np.array([0, 2]), mindspore.int32)
>>> axis = 0
>>> output = input_params.gather(input_indices, axis)
>>> print(output)
[[1.  2.  3.  4.]
 [9. 10. 11. 12.]]
>>> # case4: input_indices is a Tensor with shape (2, ) and
>>> # input_params is a Tensor with shape (3, 4) and axis is 1.
>>> input_params = Tensor(np.array([[1, 2, 3, 4], [5, 6, 7, 8], [9, 10, 11, 12]]), mindspore.float32)
>>> input_indices = Tensor(np.array([0, 2]), mindspore.int32)
>>> axis = 1
>>> output = input_params.gather(input_indices, axis)
>>> print(output)
[[1.  3.]
 [5.  7.]
 [9. 11.]]
ger(x)[源代码]

计算两个Tensor的外积,即计算此Tensor 和 x 的外积。如果此Tensor shape为 \((m,)\)x shape为 \((n,)\) , 那么输出就是一个shape为 \((m, n)\) 的Tensor。

Note

Ascend平台暂不支持float64数据格式的输入。

更多参考详见 mindspore.ops.ger()

参数:

  • x (Tensor) - 输入Tensor,数据类型为float16、float32或者float64。

返回:

Tensor,是一个与此Tensor相同数据类型的输出矩阵。当此Tensor shape为 \((m,)\)x shape为 \((n,)\) , 那么输出shape为 \((m, n)\)

支持平台:

Ascend GPU CPU

样例:

>>> x1 = Tensor([1., 2., 3., 4.], mindspore.float32)
>>> x2 = Tensor([1., 2., 3.], mindspore.float32)
>>> output = x1.ger(x2)
>>> print(output)
[[ 1.  2.  3.]
 [ 2.  4.  6.]
 [ 3.  6.  9.]
 [ 4.  8. 12.]]
hardshrink(lambd=0.5)[源代码]

Hard Shrink激活函数,按输入元素计算输出,公式定义如下:

\[\begin{split}\text{HardShrink}(x) = \begin{cases} x, & \text{ if } x > \lambda \\ x, & \text{ if } x < -\lambda \\ 0, & \text{ otherwise } \end{cases}\end{split}\]

参数:

  • lambd (float) - Hard Shrink公式定义的阈值 \(\lambda\) 。默认值:0.5。

返回:

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

异常:

  • TypeError - lambd 不是float。

  • TypeError - 原始Tensor的dtype既不是float16也不是float32。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> import mindspore as ms
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[0.5,  1,  2.0], [0.0533, 0.0776, -2.1233]]), ms.float32)
>>> print(x.hardshrink())
[[ 0.      1.      2.    ]
[ 0.      0.     -2.1233]]
property has_init

Tensor是否已经初始化。

index_fill(dim, index, value)[源代码]

index 中给定的顺序选择索引,将输入 value 值填充到当前Tensor的所有 dim 维元素。

参数:

  • dim (Union[int, Tensor]) - 填充输入Tensor的维度,要求是一个int或者数据类型为int32或int64的0维Tensor。

  • index (Tensor) - 填充输入Tensor的索引,数据类型为int32。

  • value (Union[bool, int, float, Tensor]) - 填充输入Tensor的值。如果 value 是Tensor,那么 value 要求是数据类型与当前Tensor相同的0维Tensor。否则,该值会自动转化为一个数据类型与当前Tensor相同的0维Tensor。

返回:

填充后的Tensor。shape和数据类型与当前Tensor相同。

异常:

  • TypeError - dim 的类型不是int或者Tensor。

  • TypeError - 当 dim 是Tensor时, dim 的数据类型不是int32或者int64。

  • TypeError - index 的类型不是Tensor。

  • TypeError - index 的数据类型不是int32。

  • TypeError - value 的类型不是bool、int、float或者Tensor。

  • TypeError - 当 value 是Tensor时, value 的数据类型和当前Tensor的数据类型不相同。

  • ValueError - 当 dim 是Tensor时, dim 的维度不等于0。

  • ValueError - index 的维度大于1。

  • ValueError - 当 value 是Tensor时, value 的维度不等于0。

  • RuntimeError - dim 值超出范围[-self.ndim, self.ndim - 1]。

  • RuntimeError - index 存在值超出范围[-self.shape[dim], self.shape[dim]-1]。

支持平台:

GPU

样例:

>>> import mindspore
>>> import mindspore.ops as ops
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]).astype(np.float32))
>>> index = Tensor([0, 2], mindspore.int32)
>>> value = Tensor(-2.0, mindspore.float32)
>>> y = x.index_fill(1, index, value)
>>> print(y)
[[-2. 2. -2.]
 [-2. 5. -2.]
 [-2. 8. -2.]]
init_data(slice_index=None, shape=None, opt_shard_group=None)[源代码]

获取此Tensor的数据。

Note

对于同一个Tensor,只可以调用一次 init_data 函数。

参数:

  • slice_index (int) - 参数切片的索引。在初始化参数切片的时候使用,保证使用相同切片的设备可以生成相同的Tensor。默认值:None。

  • shape (list[int]) - 切片的shape,在初始化参数切片时使用。默认值:None。

  • opt_shard_group (str) - 优化器分片组,在自动或半自动并行模式下用于获取参数的切片。默认值:None。

返回:

初始化的Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore as ms
>>> from mindspore.common.initializer import initializer, Constant
>>> x = initializer(Constant(1), [2, 2], ms.float32)
>>> out = x.init_data()
>>> print(out)
[[1. 1.]
 [1. 1.]]
soft_shrink(lambd=0.5)[源代码]

Soft Shrink激活函数,按输入元素计算输出,公式定义如下:

\[\begin{split}\text{SoftShrink}(x) = \begin{cases} x - \lambda, & \text{ if } x > \lambda \\ x + \lambda, & \text{ if } x < -\lambda \\ 0, & \text{ otherwise } \end{cases}\end{split}\]

参数:

  • lambd (float) - \(\lambda\) 应大于等于0。默认值:0.5。

返回:

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

异常:

  • TypeError - lambd 不是float。

  • TypeError - x 不是Tensor。

  • TypeError - 原始Tensor的dtype既不是float16也不是float32。

  • ValueError - lambd 小于0。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> a = Tensor([[ 0.5297,  0.7871,  1.1754], [ 0.7836,  0.6218, -1.1542]]).astype("float32")
>>> output = a.soft_shrink()
>>> print(output)
[[ 0.02979  0.287    0.676  ]
 [ 0.2837   0.1216  -0.6543 ]]
inplace_update(v, indices)[源代码]

根据 indicesv 来更新Tensor中的值。

Note

indices 只能沿着最高轴进行索引。

参数:

  • v (Tensor) - 用来更新的值。

  • indices (Union[int, tuple]) - 待更新值在原Tensor中的索引。

返回:

Tensor,更新后的Tensor。

异常:

  • TypeError - indices 不是int或tuple。

  • TypeError - indices 是元组,但是其中的元素不是int。

  • ValueError - Tensor的shape与 v 的shape不同。

支持平台:

Ascend CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> import mindspore
>>> x = Tensor(np.array([[1, 2], [3, 4], [5, 6]]), mindspore.float32)
>>> v = Tensor(np.array([[0.1, 0.2], [0.3, 0.4]]), mindspore.float32)
>>> indices = (0, 1)
>>> output = x.inplace_update(v, indices)
>>> print(output)
[[0.1 0.2]
 [0.3 0.4]
 [5.  6. ]]
isclose(x2, rtol=1e-05, atol=1e-08, equal_nan=False)[源代码]

返回一个布尔型Tensor,表示当前Tensor与 x2 的对应元素的差异是否在容忍度内相等。

参数:

  • x2 (Tensor) - 对比的第二个输入,支持的类型有float32,float16,int32。

  • rtol (float, optional) - 相对容忍度。默认值:1e-05。

  • atol (float, optional) - 绝对容忍度。默认值:1e-08。

  • equal_nan (bool, optional) - IsNan的输入,任意维度的Tensor。默认值:False。

返回:

Tensor,shape与广播后的shape相同,数据类型是布尔型。

异常:

  • TypeError - 当前Tensor和 x2 中的任何一个不是Tensor。

  • TypeError - 当前Tensor和 x2 的数据类型不是float16、float32或int32之一。

  • TypeError - atolrtol 中的任何一个不是float。

  • TypeError - equal_nan 不是bool。

  • TypeError - 当前Tensor和 x2 的数据类型不同。

  • ValueError - 当前Tensor和 x2 无法广播。

  • ValueError - atolrtol 中的任何一个小于零。

支持平台:

CPU

样例:

>>> input = Tensor(np.array([1.3, 2.1, 3.2, 4.1, 5.1]), mindspore.float16)
>>> other = Tensor(np.array([1.3, 3.3, 2.3, 3.1, 5.1]), mindspore.float16)
>>> output = input.isclose(other)
>>> print(output)
[ True False False False  True]
isfinite()[源代码]

判断输入数据每个位置上的值是否是有限数。

返回:

Tensor,输出的shape与输入相同,数据类型为bool。

异常:

  • TypeError - 如果当前Tensor不是Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> a = Tensor(np.array([np.log(-1), 1, np.log(0)]), mindspore.float32)
>>> output = a.isfinite()
>>> print(output)
[False True False]
item(index=None)[源代码]

获取Tensor中指定索引的元素。

Note

Tensor.item返回的是Tensor标量,而不是Python标量。

参数:

  • index (Union[None, int, tuple(int)]) - Tensor的索引。默认值:None。

返回:

Tensor标量,dtype与原始Tensor的相同。

异常:

  • ValueError - index 的长度不等于Tensor的ndim。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1,2,3],[4,5,6]], dtype=np.float32))
>>> x = x.item((0,1))
>>> print(x)
2.0
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.]]
property itemsize

返回一个Tensor元素的长度(以字节为单位)。

lerp(end, weight)[源代码]

基于某个浮点数Scalar或权重Tensor的值, 计算当前Tensor和 end Tensor之间的线性插值。

如果参数 weight 是一个Tensor,那么另两个输入的维度信息可以被广播到当前Tensor。 如果参数 weight 是一个Scalar, 那么 end 的维度信息可以被广播到当前Tensor。

参数:

  • end (Tensor) - 进行线性插值的Tensor结束点,其数据类型必须为float16或者float32。

  • weight (Union[float, Tensor]) - 线性插值公式的权重参数。当为Scalar时,其数据类型为float,当为Tensor时,其数据类型为float16或者float32。

返回:

返回新的Tensor,其数据类型和维度必须和输入中的当前Tensor保持一致。

异常:

  • TypeError - 如果 end 不是Tensor。

  • TypeError - 如果 weight 不是float类型Scalar或者Tensor。

  • TypeError - 如果 end 的数据类型不是float16或者float32。

  • TypeError - 如果 weight 为Tensor且 weight 不是float16或者float32。

  • TypeError - 如果当前Tensor和 end 的数据类型不一致。

  • TypeError - 如果 weight 为Tensor且 endweight 和当前Tensor数据类型不一致。

  • ValueError - 如果 end 的维度信息无法相互广播到当前Tensor。

  • ValueError - 如果 weight 为Tensor且 weight 的维度信息无法广播到当前Tensor。

支持平台:

Ascend CPU

样例:

>>> start = Tensor(np.array([1., 2., 3., 4.]), mindspore.float32)
>>> end = Tensor(np.array([10., 10., 10., 10.]), mindspore.float32)
>>> output = start.lerp( end, 0.5)
>>> print(output)
[5.5 6. 6.5 7. ]
log1p()[源代码]

对当前Tensor逐元素加一后计算自然对数。

\[out_i = {log_e}(x_i + 1)\]

返回:

Tensor,与 x 的shape相同。

异常:

  • TypeError - x 不是Tensor。

  • TypeError - x 的数据类型非float16或float32。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([1.0, 2.0, 4.0]), mindspore.float32)
>>> output = x.log1p()
>>> print(output)
[0.6931472 1.0986123 1.609438 ]
norm(axis, p=2, keep_dims=False, epsilon=1e-12)[源代码]

返回给定Tensor的矩阵范数或向量范数。

\[output = sum(abs(input)**p)**(1/p)\]

参数:

  • axis (Union[int, list, tuple]) - 指定要计算范数的输入维度。

  • p (int) - 范数的值。默认值:2。 p 大于等于0。

  • keep_dims (bool) - 输出Tensor是否保留原有的维度。默认值:False。

  • epsilon (float) - 用于保持数据稳定性的常量。默认值:1e-12。

返回:

Tensor,其数据类型与当前Tensor相同,其维度信息取决于 axis 轴以及参数 keep_dims 。例如如果输入的大小为 (2,3,4) 轴为 [0,1] ,输出的维度为 (4,)

异常:

  • TypeError - 当前Tensor的数据类型不是float16或者float32。

  • TypeError - axis 不是int,tuple或者list。

  • TypeError - p 不是int。

  • TypeError - axis 是tuple或者list但其元素不是int。

  • TypeError - keep_dims 不是bool。

  • TypeError - epsilon 不是float。

  • ValueError - axis 的元素超出范围 (-len(input_x.shape), len(input_x.shape)) ,其中 input_x 指当前Tensor。

  • ValueError - axis 的维度rank大于当前Tensor的维度rank。

支持平台:

Ascend GPU CPU

样例:

>>> input_x = Tensor(np.array([[[1.0, 2.0], [3.0, 4.0]], [[5.0, 6.0], [7.0, 8.0]]]).astype(np.float32))
>>> output = input_x.norm([0, 1], p=2)
>>> print(output)
[ 9.165152 10.954452]
masked_fill(mask, value)[源代码]

将掩码位置为True的位置填充指定的值。该Tensor和 mask 的shape需相同或可广播。

参数:

  • mask (Tensor[bool]) - mask矩阵,值为bool类型的Tensor。

  • value (Union[float, Tensor]) - 填充值,其数据类型与该Tensor相同。

返回:

Tensor,shape和dtype与该Tensor相同。

异常:

  • TypeError - mask 不是Tensor。

  • TypeError - mask 的数据类型不是bool。

  • ValueError - 该Tensor和 mask 的shape不可广播。

  • TypeError - 该Tensor 或 value 的数据类型不是float16、float32、int8、或int32。

  • TypeError - value 的数据类型与该Tensor不同。

  • TypeError - value 既不是float也不是Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> a = Tensor(np.arange(4)).astype('float32')
>>> print(a)
[0. 1. 2. 3.]
>>> mask = Tensor([False, False, True, True])
>>> print(a.masked_fill(mask, 0.0))
[0. 1. 0. 0.]
masked_select(mask)[源代码]

返回一个一维张量,其中的内容是此张量中对应于 mask 张量中True位置的值。mask 张量的shape与此张量的shape不需要一样,但必须符合广播规则。

参数:

  • mask (Tensor[bool]) - 值为bool类型的张量。

返回:

一个一维张量,类型与此张量相同。

异常:

  • TypeError - mask 不是bool类型的Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([1, 2, 3, 4]), mindspore.int64)
>>> mask = Tensor(np.array([1, 0, 1, 0]), mindspore.bool_)
>>> output = x.masked_select(mask)
>>> print(output)
[1 3]
inv()[源代码]

计算当前Tensor的倒数。

\[out_i = \frac{1}{x_{i} }\]

其中 x 表示当前Tensor。

返回:

Tensor,shape和类型与当前Tensor相同。

异常:

  • TypeError - 当前Tensor的数据类型不为float16、float32或int32。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([0.25, 0.4, 0.31, 0.52]), mindspore.float32)
>>> output = x.inv()
>>> print(output)
[4.        2.5       3.2258065 1.923077 ]
invert()[源代码]

按位翻转当前Tensor。

\[out_i = \sim x_{i}\]

其中 x 表示当前Tensor。

返回:

Tensor,shape和类型与当前Tensor相同。

异常:

  • TypeError - 当前Tensor的数据类型不为int16或uint16。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([25, 4, 13, 9]), mindspore.int16)
>>> output = x.invert()
>>> print(output)
[-26 -5 -14 -10]
max(axis=None, keepdims=False, initial=None, where=True)[源代码]

返回Tensor的最大值或轴方向上的最大值。

参数:

  • axis (Union[None, int, list, tuple of ints], optional) - 轴,在该轴方向上进行操作。默认情况下,使用扁平输入。如果该参数为整数元组,则在多个轴上选择最大值,而不是在单个轴或所有轴上进行选择。默认值:None。

  • keepdims (bool, optional) - 如果这个参数为True,被删去的维度保留在结果中,且维度大小设为1。有了这个选项,结果就可以与输入数组进行正确的广播运算。默认值:False。

  • initial (scalar, optional) - 输出元素的最小值。如果对空切片进行计算,则该参数必须设置。默认值:None。

  • where (bool Tensor, optional) - 一个bool数组,被广播以匹配数组维度和选择包含在降维中的元素。如果传递了一个非默认值,则还必须提供初始值。默认值:True。

返回:

Tensor或标量,输入Tensor的最大值。如果 axis 为None,则结果是一个标量值。如果提供了 axis ,则结果是Tensor ndim - 1维度的一个数组。

异常:

  • TypeError - 参数具有前面未指定的类型。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> a = Tensor(np.arange(4).reshape((2, 2)).astype('float32'))
>>> output = a.max()
>>> print(output)
3.0
mean(axis=(), keep_dims=False)[源代码]

返回指定维度上所有元素的均值,并降维。

参数:

  • axis (Union[None, int, tuple(int), list(int)]) - 要减少的维度。默认值: (),缩小所有维度。当 axis 为int、tuple(int)或list(int)时,记Tensor的维度为dim,则其取值范围为[-dim, dim)。

  • keep_dims (bool) - 计算结果是否保留维度。默认值:False。

返回:

与输入的张量具有相同的数据类型的Tensor。

  • 如果 axis 为(),且 keep_dims 为False,则输出一个0维Tensor,表示输入Tensor中所有元素的均值。

  • 如果 axis 为int,取值为1,并且 keep_dims 为False,则输出的shape为 \((x_0, x_2, ..., x_R)\)

  • 如果 axis 为tuple(int)或list(int),取值为(1, 2),并且 keep_dims 为False,则输出Tensor的shape为 \((x_0, x_3, ..., x_R)\)

异常:

  • TypeError - 如果 axis 不是以下数据类型之一:int、tuple 或 list。

  • TypeError - 如果 keep_dims 不是bool类型。

  • ValueError - 如果 axis 超出范围。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([1, 2, 3], dtype=np.float32))
>>> output = input_x.mean()
>>> print(output)
2.0
amin(axis=(), keep_dims=False)[源代码]

默认情况下,使用指定维度的最小值代替该维度的其他元素,以移除该维度。也可仅缩小该维度大小至1。 keep_dims 控制输出和输入的维度是否相同。

参数:

  • axis (Union[None, int, tuple(int), list(int)]) - 要减少的维度。默认值: (),缩小所有维度。只允许常量值。当 axis 为int、tuple(int)或list(int)时,记Tensor的维度为dim,则其取值范围为[-dim, dim)。

  • keep_dims (bool) - 如果为True,则保留缩小的维度,大小为1。否则移除维度。默认值:False。

返回:

与输入的张量具有相同的数据类型的Tensor。

  • 如果 axis 为(),且 keep_dims 为False,则输出一个0维Tensor,表示输入Tensor中所有元素的最小值。

  • 如果 axis 为int,取值为1,并且 keep_dims 为False,则输出的shape为 \((x_0, x_2, ..., x_R)\)

  • 如果 axis 为tuple(int)或list(int),取值为(1, 2),并且 keep_dims 为False,则输出Tensor的shape为 \((x_0, x_3, ..., x_R)\)

异常:

  • TypeError - 如果 axis 不是以下数据类型之一:int、tuple 或 list。

  • TypeError - 如果 keep_dims 不是bool类型。

  • ValueError - 如果 axis 超出范围。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([1, 2, 3], dtype=np.float32))
>>> output = input_x.amin()
>>> print(output)
1.0
amax(axis=(), keep_dims=False)[源代码]

默认情况下,使用指定维度的最大值代替该维度的其他元素,以移除该维度。也可仅缩小该维度大小至1。 keep_dims 控制输出和输入的维度是否相同。

参数:

  • axis (Union[None, int, tuple(int), list(int)]) - 要减少的维度。默认值: (),缩小所有维度。只允许常量值。当 axis 为int、tuple(int)或list(int)时,记Tensor的维度为dim,则其取值范围为[-dim, dim)。

  • keep_dims (bool) - 如果为True,则保留缩小的维度,大小为1。否则移除维度。默认值:False。

返回:

与输入的张量具有相同的数据类型的Tensor。

  • 如果 axis 为(),且 keep_dims 为False,则输出一个0维Tensor,表示输入Tensor中所有元素的最大值。

  • 如果 axis 为int,取值为1,并且 keep_dims 为False,则输出的shape为 \((x_0, x_2, ..., x_R)\)

  • 如果 axis 为tuple(int)或list(int),取值为(1, 2),并且 keep_dims 为False,则输出Tensor的shape为 \((x_0, x_3, ..., x_R)\)

异常:

  • TypeError - 如果 axis 不是以下数据类型之一:int、tuple 或 list。

  • TypeError - 如果 keep_dims 不是bool类型。

  • ValueError - 如果 axis 超出范围。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([1, 2, 3], dtype=np.float32))
>>> output = input_x.amax()
>>> print(output)
3.0
prod(axis=(), keep_dims=False)[源代码]

默认情况下,通过将维度中的所有元素相乘来减少张量的维度。并且还可以沿轴减小“x”的维度。通过控制 keep_dims 判断输出和输入的维度是否相同。

参数:

  • axis (Union[None, int, tuple(int), list(int)]) - 计算prod的维度。当 axis 为None或空元组时,计算所有维度。当 axis 为int、tuple(int)或list(int)时,记Tensor的维度为dim,则其取值范围为[-dim, dim)。默认值:()。

  • keep_dims (bool) - 计算结果是否保留维度。默认值:False。

返回:

与输入的张量具有相同的数据类型的Tensor。

  • 如果 axis 为(),且 keep_dims 为False,则输出一个0维Tensor,表示输入Tensor中所有元素的乘积。

  • 如果 axis 为int,取值为1,并且 keep_dims 为False,则输出的shape为 \((x_0, x_2, ..., x_R)\)

  • 如果 axis 为tuple(int)或list(int),取值为(1, 2),并且 keep_dims 为False,则输出Tensor的shape为 \((x_0, x_3, ..., x_R)\)

异常:

  • TypeError - 如果 axis 不是以下数据类型之一:int、tuple 或 list。

  • TypeError - 如果 keep_dims 不是bool类型。

  • ValueError - 如果 axis 超出范围。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([1, 2, 3], dtype=np.float32))
>>> output = input_x.prod()
>>> print(output)
6.0
min(axis=None, keepdims=False, initial=None, where=True)[源代码]

返回Tensor的最小值或轴方向上的最小值。

参数:

  • axis (Union[None, int, list, tuple of ints], optional) - 轴,在该轴方向上进行操作。默认情况下,使用扁平输入。如果该参数为整数元组,则在多个轴上选择最小值,而不是在单个轴或所有轴上进行选择。默认值:None。

  • keepdims (bool, optional) - 如果这个参数为True,被删去的维度保留在结果中,且维度大小设为1。有了这个选项,结果就可以与输入数组进行正确的广播运算。默认值:False。

  • initial (scalar, optional) - 输出元素的最大值。如果对空切片进行计算,则该参数必须设置。默认值:None。

  • where (bool Tensor, optional) - 一个布尔数组,被广播以匹配数组维度和选择包含在降维中的元素。如果传递了一个非默认值,则还必须提供初始值。默认值:True。

返回:

Tensor或标量,输入Tensor的最小值。如果轴为None,则结果为一个标量值。如果提供了 axis ,则结果是Tensor.ndim - 1维度的一个数组。

异常:

  • TypeError - 参数具有前面未指定的类型。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> import mindspore.numpy as np
>>> a = Tensor(np.arange(4).reshape((2,2)).astype('float32'))
>>> output = a.min()
>>> print(output)
0.0
narrow(axis, start, length)[源代码]

沿指定轴,指定起始位置获取指定长度的Tensor。

参数:

  • axis (int) - 指定的轴。

  • start (int) - 指定的起始位置。

  • length (int) - 指定的长度。

返回:

Tensor。

异常:

  • TypeError - axis不是int类型。

  • TypeError - start不是int类型。

  • TypeError - length不是int类型。

  • ValueError - axis取值不在[0, ndim-1]范围内。

  • ValueError - start取值不在[0, shape[axis]-1]范围内。

  • ValueError - start+length超出Tensor的维度范围shape[axis]-1。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore
>>> from mindspore import Tensor
>>> x = Tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]], mindspore.int32)
>>> output = x.narrow(0, 0, 2)
>>> print(output)
[[ 1 2 3]
 [ 4 5 6]]
>>> output = x.narrow(1, 1, 2)
>>> print(output)
[[ 2 3]
 [ 5 6]
 [ 8 9]]
property nbytes

返回Tensor占用的总字节数。

property ndim

返回Tensor维度的数量。

nonzero()[源代码]

计算x中非零元素的下标。

返回:

Tensor,维度为2,类型为int64,表示输入中所有非零元素的下标。

支持平台:

GPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[[1,  0], [-5, 0]]]), mindspore.int32)
>>> output = x.nonzero()
>>> print(output)
[[0 0 0]
 [0 1 0]]
pow(power)[源代码]

计算Tensor中每个元素的 power 次幂。

\[out_{i} = x_{i} ^{ y_{i}}\]

Note

  • Tensor和 power 遵循 隐式类型转换规则 ,使数据类型保持一致。

  • 当前的Tensor和 power 的数据类型不能同时是bool,并保证其shape可以广播。

参数:

  • power (Union[Tensor, number.Number, bool]) - 幂值,是一个number.Number或bool值,或数据类型为number或bool_的Tensor。

返回:

Tensor,shape与广播后的shape相同,数据类型为 Tensorpower 中精度较高的类型。

异常:

  • TypeError - power 不是Tensor、number.Number或bool。

  • ValueError - 当Tensor和 power 都为Tensor时,它们的shape不相同。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([1.0, 2.0, 4.0]), mindspore.float32)
>>> y = 3.0
>>> output = x.pow(y)
>>> print(output)
[ 1.  8. 64.]
>>>
>>> x = Tensor(np.array([1.0, 2.0, 4.0]), mindspore.float32)
>>> y = Tensor(np.array([2.0, 4.0, 3.0]), mindspore.float32)
>>> output = x.pow(y)
>>> print(output)
[ 1. 16. 64.]
ptp(axis=None, keepdims=False)[源代码]

该函数名称是”peak to peak”的缩写。计算沿着axis的最大值与最小值的差值。

Note

不支持NumPy参数 dtypeout

参数:

  • axis (Union[None, int, tuple(int)]) - 轴,在轴方向上可以计算范围。默认计算扁平数组的方差。默认值:None。

  • keepdims (bool) - 如果设为True,被删去的维度保留在结果中,且维度大小设为1。有了这个选项,结果将针对输入数组正确传递。默认值为False。

返回:

Tensor。

异常:

  • TypeError - self 不是Tensor,或者 axiskeepdims 具有前面未指定的类型。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> x = Tensor([[4.0, 9.0, 2.0, 10.0], [6.0, 9.0, 7.0, 12.0]]).astype("float32")
>>> print(x.ptp(axis=1))
[8. 6.]
>>> print(x.ptp(axis=0))
[2. 0. 5. 2.]
ravel()[源代码]

返回一个展开的一维Tensor。

返回:

一维Tensor,含有与输入相同的元素。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((2,3,4), dtype=np.float32))
>>> output = x.ravel()
>>> print(output.shape)
(24,)
renorm(p, dim, maxnorm)[源代码]

沿维度 dim 重新规范Tensor的子张量,并且每个子张量的p范数不超过给定的最大范数 maxnorm 。 如果子张量的p范数小于 maxnorm ,则当前子张量不需要修改;否则该子张量需要修改为对应位置的原值除以该子张量的p范数,然后再乘上 maxnorm

参数:

  • p (int) - 范数计算的幂。

  • dim (int) - 获得子张量的维度。

  • maxnorm (float32) - 给定的最大范数。

返回:

Tensor,shape和type与输入Tensor一致。

异常:

  • TypeError - p 不是int类型。

  • TypeError - dim 不是int类型。

  • TypeError - maxnorm 不是float32类型。

  • ValueError - p 小于等于0。

支持平台:

Ascend CPU GPU

样例:

>>> x = Tensor(np.array([[1, 1, 1], [2, 2, 2], [3, 3, 3]]), mindspore.float32)
>>> y = x.renorm(p=1, dim=0, maxnorm=5.)
>>> print(y)
[[1.       1.        1.        ]
[1.6666666 1.6666666 1.6666666 ]
[1.6666667 1.6666667 1.6666667 ]]
repeat(repeats, axis=None)[源代码]

对数组中的元素进行重复复制。

参数:

  • repeats (Union[int, tuple, list]) - 每个元素的重复次数,repeats 被广播以适应指定轴的shape。

  • axis (int, optional) - 轴方向上的重复值。默认情况下,使用展开的输入Tensor,并返回一个展开的输出Tensor。

返回:

Tensor,除了维度外,与输入Tensor具有相同的shape。

异常:

  • ValueError - 维度超出范围。

  • TypeError - 参数类型不匹配。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array(3))
>>> print(x.repeat(4))
[3 3 3 3]
>>> x = Tensor(np.array([[1, 2],[3, 4]]))
>>> print(x.repeat(2))
[1 1 2 2 3 3 4 4]
>>> print(x.repeat(3, axis=1))
[[1 1 1 2 2 2]
[3 3 3 4 4 4]]
>>> print(x.repeat([1,2], axis=0))
[[1 2]
[3 4]
[3 4]]
reshape(*shape)[源代码]

不改变数据的情况下,将Tensor的shape改为输入的新shape。

参数:

shape (Union[int, tuple(int), list(int)]) - 新的shape应与原来的shape兼容。如果参数值为整数,则结果是该长度的一维数组。shape的维度可以为-1。在这种情况下,将根据数组的长度和剩下的维度计算出该值。

返回:

Tensor,具有新shape的Tensor。

异常:

  • TypeError - 新shape不是整数、列表或元组。

  • ValueError - 新shape与原来Tensor的shape不兼容。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> from mindspore import dtype as mstype
>>> x = Tensor([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]], dtype=mstype.float32)
>>> output = x.reshape((3, 2))
>>> print(output)
[[-0.1  0.3]
[ 3.6  0.4]
[ 0.5 -3.2]]
resize(*new_shape)[源代码]

将Tensor改为输入的新shape, 并将不足的元素补0。

Note

此方法不更改输入数组的大小,也不返回NumPy中的任何内容,而是返回一个具有输入大小的新Tensor。不支持Numpy参数 refcheck

参数:

  • new_shape (Union[ints, tuple of ints]) - 指定Tensor的新shape。

返回:

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))
>>> y = x.resize(3, 3)
>>> print(y)
[[1. 2. 3.]
[4. 5. 6.]
[0. 0. 0.]]
>>> y = x.resize(2, 2)
>>> print(y)
[[1. 2.]
[3. 4.]]
round()[源代码]

将Tensor进行四舍五入到最接近的整数数值。

返回:

Tensor,shape和数据类型与原Tensor相同。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([0.8, 1.5, 2.3, 2.5, -4.5]), mindspore.float32)
>>> output = x.round()
>>> print(output)
[ 1.  2.  2.  2. -4.]
select(condition, y)[源代码]

根据条件判断Tensor中的元素的值,来决定输出中的相应元素是从当前Tensor(如果元素值为True)还是从 y (如果元素值为False)中选择。

该算法可以被定义为:

\[\begin{split}out_i = \begin{cases} tensor_i, & \text{if } condition_i \\ y_i, & \text{otherwise} \end{cases}\end{split}\]

参数:

  • condition (Tensor[bool]) - 条件Tensor,决定选择哪一个元素。shape与当前的Tensor相同。

  • y (Union[Tensor, int, float]) - 如果y是一个Tensor,那么shape与当前Tensor相同。如果y是int或者float, 那么将会被转化为int32或者float32类型,并且被广播为与当前Tensor相同的shape。

返回:

Tensor,与当前Tensor的shape相同。

异常:

  • TypeError - y 不是Tensor、int或者float。

  • ValueError - 输入的shape不相同。

支持平台:

Ascend GPU CPU

样例:

>>> # 1) y is Tensor
>>>
>>> cond = Tensor([True, False])
>>> x = Tensor([2,3], mindspore.float32)
>>> y = Tensor([1,2], mindspore.float32)
>>> output = x.select(cond, y)
>>> print(output)
[2. 2.]
>>> # 2) y is a float
>>> cond = Tensor([True, False])
>>> x = Tensor([2,3], mindspore.float32)
>>> y = 2.0
>>> output = x.select(cond, y)
>>> print(output)
[2. 2.]
property shape

返回Tensor的shape。

property size

返回Tensor中的元素总数。

squeeze(axis=None)[源代码]

从Tensor中删除shape为1的维度。

参数:

  • axis (Union[None, int, list(int), tuple(int)], optional) - 选择shape中长度为1的条目的子集。如果选择shape条目长度大于1的轴,则报错。默认值为None。

返回:

Tensor,删除了长度为1的维度的全部子集或一个子集。

异常:

  • TypeError - 输入的参数类型有误。

  • ValueError - 指定维度的shape大于1。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((1,2,2), dtype=np.float32))
>>> print(x)
[[[1. 1.]
[1. 1.]]]
>>> print(x.shape)
(1, 2, 2)
>>> y = x.squeeze()
>>> print(y)
[[1. 1.]
[1. 1.]]
>>> print(y.shape)
(2, 2)
>>> y = x.squeeze(axis=0)
>>> print(y)
[[1. 1.]
[1. 1.]]
>>> print(y.shape)
(2, 2)
std(axis=None, ddof=0, keepdims=False)[源代码]

计算指定维度的标准差。 标准差是方差的算术平方根,如:\(std = sqrt(mean(abs(x - x.mean())**2))\)

返回标准差。默认情况下计算展开数组的标准差,否则在指定维度上计算。

Note

不支持NumPy参数 dtypeoutwhere

参数:

  • axis (Union[None, int, tuple(int)]) - 在该维度上计算标准差。默认值:None 。如果为 None ,则计算展开数组的标准偏差。

  • ddof (int) - δ自由度。计算中使用的除数是 \(N - ddof\) ,其中 \(N\) 表示元素的数量。默认值:0。

  • keepdims - 默认值:False

返回:

含有标准差数值的Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([1, 2, 3, 4], dtype=np.float32))
>>> output = input_x.std()
>>> print(output)
1.118034
property strides

Tensor上每个维度跨度的字节元组。

sum(axis=None, dtype=None, keepdims=False, initial=None)[源代码]

返回指定维度上数组元素的总和。

Note

不支持NumPy参数 outwherecastingordersuboksignatureextobj

参数:

  • axis (Union[None, int, tuple(int)]) - 指定维度,在该维度方向上进行求和运算。默认值:None。如果参数值为None,会计算输入数组中所有元素的和。如果axis为负数,则从最后一维开始往第一维计算。如果axis为整数元组,会对该元组指定的所有轴方向上的元素进行求和。

  • dtype (mindspore.dtype, optional) - 默认值为None。会覆盖输出Tensor的dtype。

  • keepdims (bool) - 如果这个参数为True,被删去的维度保留在结果中,且维度大小设为1。有了这个选项,结果就可以与输入数组进行正确的广播运算。如果设为默认值,那么 keepdims 不会被传递给ndarray子类的sum方法。但是任何非默认值都会被传递。如果子类的方法未实现 keepdims ,则引发异常。默认值:False。

  • initial (scalar) - 初始化的起始值。默认值:None。

返回:

Tensor。具有与输入相同shape的Tensor,删除了指定的轴。如果输入Tensor是0维数组,或axis为None时,返回一个标量。

异常:

  • TypeError - input不是Tensor,axis 不是整数或整数元组,keepdims 不是整数,或者 initial 不是标量。

  • ValueError - 任意轴超出范围或存在重复的轴。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([-1, 0, 1]).astype(np.float32))
>>> print(input_x.sum())
0.0
>>> input_x = Tensor(np.arange(10).reshape(2, 5).astype(np.float32))
>>> print(input_x.sum(axis=1))
[10. 35.]
swapaxes(axis1, axis2)[源代码]

交换Tensor的两个维度。

参数:

  • axis1 (int) - 第一个维度。

  • axis2 (int) - 第二个维度。

返回:

转化后的Tensor,与输入具有相同的数据类型。

异常:

  • TypeError - axis1axis2 不是整数。

  • ValueError - axis1axis2 不在 [-ndim, ndim-1] 范围内。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((2,3,4), dtype=np.float32))
>>> output = x.swapaxes(0, 2)
>>> print(output.shape)
(4,3,2)
take(indices, axis=None, mode='clip')[源代码]

在指定维度上获取Tensor中的元素。

参数:

  • indices (Tensor) - 待提取的值的shape为 (Nj…) 的索引。

  • axis (int, optional) - 在指定维度上选择值。默认情况下,使用展开的输入数组。默认值:None。

  • mode (‘raise’, ‘wrap’, ‘clip’, optional)

    • raise:抛出错误。

    • wrap:绕接。

    • clip:裁剪到范围。 clip 模式意味着所有过大的索引都会被在指定轴方向上指向最后一个元素的索引替换。注:这将禁用具有负数的索引。默认值:clip

返回:

Tensor,索引的结果。

异常:

  • ValueError - axis 超出范围,或 mode 被设置为’raise’、’wrap’和’clip’以外的值。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> a = Tensor(np.array([4, 3, 5, 7, 6, 8]))
>>> indices = Tensor(np.array([0, 1, 4]))
>>> output = a.take(indices)
>>> print(output)
[4 3 6]
tan()[源代码]

返回每个元素的正切值。

\[out_i = tan(x_i)\]

返回:

Tensor。

异常:

  • TypeError - 当前输入不是Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> from mindspore import Tensor
>>> a = Tensor([-1.0, 0.0, 1.0]).astype("float32")
>>> output = a.tan()
>>> print(output)
[-1.5574081 0. 1.5574081]
scatter_add(indices, updates)[源代码]

根据指定的更新值和输入索引,通过加法进行运算,将结果赋值到输出Tensor中。当同一索引有不同值时,更新的结果将是所有值的总和。此操作几乎等同于使用 mindspore.ops.ScatterNdAdd ,只是更新后的结果是通过算子output返回,而不是直接原地更新input。

indices 的最后一个轴是每个索引向量的深度。对于每个索引向量, updates 中必须有相应的值。updates 的shape应该等于 input_x[indices] 的shape,其中 input_x 指当前Tensor。有关更多详细信息,请参见使用用例。

Note

GPU平台上,如果 indices 的某些值超出范围,则相应的 updates 不会更新到 input_x ,而不是抛出索引错误;CPU平台上直接抛出索引错误;Ascend平台不支持越界检查,若越界可能会造成未知错误。

参数:

  • indices (Tensor) - Tensor的索引,数据类型为int32或int64。其rank至少为2。

  • updates (Tensor) - 指定与本Tensor相加操作的Tensor,其数据类型与该Tensor相同。 updates.shape 应等于 indices.shape[:-1] + self.shape[indices.shape[-1]:]

返回:

Tensor,shape和数据类型与原Tensor相同。

异常:

  • TypeError - indices 的数据类型既不是int32,也不是int64。

  • ValueError - Tensor的shape长度小于 indices 的shape的最后一个维度。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]).astype('float32'))
>>> indices = Tensor(np.array([[0, 0], [0, 0]]).astype('int32'))
>>> updates = Tensor(np.array([1.0, 2.2]).astype('float32'))
>>> output = x.scatter_add(indices, updates)
>>> print(output)
[[ 3.1  0.3  3.6]
[ 0.4  0.5 -3.2]]
scatter_div(indices, updates)[源代码]

根据索引,通过相除运算得到输出Tensor的值。更新后的结果是通过算子output返回,而不是直接原地更新当前Tensor。

indices 的最后一个轴是每个索引向量的深度。对于每个索引向量, updates 中必须有相应的值。 updates 的shape应该等于 input_x[indices] 的shape。其中 input_x 指当前Tensor。 有关更多详细信息,请参见使用用例。

Note

  • 如果 indices 的某些值超出范围,则相应的 updates 不会更新为当前Tensor,而不是抛出索引错误。

  • 算子无法处理除0异常,用户需保证 updates 中没有0值。

参数:

  • indices (Tensor) - 该Tensor的索引,数据类型为int32或int64。其rank至少为2。

  • updates (Tensor) - 指定与当前Tensor相加操作的Tensor,其数据类型与输入相同。 updates.shape 应等于 indices.shape[:-1] + input_x.shape[indices.shape[-1]:] ,其中 input_x 指当前Tensor。

返回:

Tensor,shape和数据类型与该Tensor相同。

异常:

  • TypeError - indices 的数据类型不是int32,也不是int64。

  • ValueError - Tensor的shape长度小于 indices 的shape的最后一个维度。

支持平台:

GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]).astype('float32'))
>>> indices = Tensor(np.array([[0, 0], [0, 0]]).astype('int32'))
>>> updates = Tensor(np.array([1.0, 2.0]).astype('float32'))
>>> output = input_x.scatter_div(indices, updates)
>>> print(output)
[[-0.05  0.3  3.6  ]
 [ 0.4   0.5  -3.2 ]]
scatter_min(indices, updates)[源代码]

根据指定的更新值和输入索引,通过最小值运算,将结果赋值到输出Tensor中。

索引的最后一个轴是每个索引向量的深度。对于每个索引向量, updates 中必须有相应的值。 updates 的shape应该等于 input_x[indices] 的shape。有关更多详细信息,请参见下方样例。

Note

如果 indices 的某些值超出范围,则相应的 updates 不会更新到 input_x ,而不是抛出索引错误。

参数:

  • indices (Tensor) - Tensor的索引,数据类型为int32或int64。其rank至少为2。

  • updates (Tensor) - 指定与本Tensor做最小值运算的Tensor,其数据类型与该Tensor相同。 updates.shape 应等于 indices.shape[:-1] + self.shape[indices.shape[-1]:]

返回:

Tensor,shape和数据类型与原Tensor相同。

异常:

  • TypeError - indices 的数据类型既不是int32,也不是int64。

  • ValueError - Tensor的shape长度小于 indices 的shape的最后一个维度。

支持平台:

GPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]).astype('float32'))
>>> indices = Tensor(np.array([[0, 0], [0, 0]]).astype('int32'))
>>> updates = Tensor(np.array([1.0, 2.2]).astype('float32'))
>>> output = x.scatter_min(indices, updates)
>>> print(output)
[[ -0.1  0.3  3.6]
[ 0.4  0.5 -3.2]]
scatter_max(indices, updates)[源代码]

根据指定的更新值和输入索引,通过最大值运算,输出结果以Tensor形式返回。

索引的最后一个轴是每个索引向量的深度。对于每个索引向量, updates 中必须有相应的值。 updates 的shape应该等于 input_x[indices] 的shape。有关更多详细信息,请参见下方样例。

Note

如果 indices 的某些值超出范围,则不会更新相应的 updates,同时也不会抛出索引错误。

参数:

  • indices (Tensor) - Tensor的索引,数据类型为int32或int64的。其rank必须至少为2。

  • updates (Tensor) - 指定与本Tensor做最大值运算的Tensor,其数据类型与该Tensor相同。 updates.shape 应等于 indices.shape[:-1] + self.shape[indices.shape[-1]:]

返回:

Tensor,shape和数据类型与原Tensor相同。

异常:

  • TypeError - indices 的数据类型既不是int32,也不是int64。

  • ValueError - Tensor的shape长度小于 indices 的shape的最后一个维度。

支持平台:

GPU

样例:

>>> input_x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]), mindspore.float32)
>>> indices = Tensor(np.array([[0, 0], [0, 0]]), mindspore.int32)
>>> updates = Tensor(np.array([1.0, 2.2]), mindspore.float32)
>>> # Next, demonstrate the approximate operation process of this operator:
>>> # 1, indices[0] = [0, 0], indices[1] = [0, 0]
>>> # 2, And input_x[0, 0] = -0.1
>>> # 3, So input_x[indices] = [-0.1, -0.1]
>>> # 4, Satisfy the above formula: input_x[indices].shape=(2) == updates.shape=(2)
>>> op = ops.TensorScatterMax()
>>> # 5, Perform the max operation for the first time:
>>> #      first_input_x = Max(input_x[0][0], updates[0]) = [[1.0, 0.3, 3.6], [0.4, 0.5, -3.2]]
>>> # 6, Perform the max operation for the second time:
>>> #      second_input_x = Max(input_x[0][0], updates[1]) = [[2.2, 0.3, 3.6], [0.4, 0.5, -3.2]]
>>> output = op(input_x, indices, updates)
>>> print(output)
[[ 2.2  0.3  3.6]
[ 0.4  0.5 -3.2]]
scatter_mul(indices, updates)[源代码]

根据指定的索引,通过乘法进行计算,将结果赋值到输出Tensor中。更新后的结果是通过算子output返回,而不是直接原地更新当前Tensor。

indices 的最后一个轴是每个索引向量的深度。对于每个索引向量, updates 中必须有相应的值。 updates 的shape应该等于 input_x[indices] 的shape。其中 input_x 指当前Tensor。 有关更多详细信息,请参见使用用例。

Note

  • 如果 indices 的某些值超出范围,则相应的 updates 不会更新为当前Tensor,而不是抛出索引错误。

参数:

  • indices (Tensor) - 该Tensor的索引,数据类型为int32或int64的。其rank必须至少为2。

  • updates (Tensor) - 指定与当前Tensor相加操作的Tensor,其数据类型与输入相同。updates.shape应等于 indices.shape[:-1] + input_x.shape[indices.shape[-1]:], 其中 input_x 代指当前Tensor本身。

返回:

Tensor,shape和数据类型与该Tensor相同。

异常:

  • TypeError - indices 的数据类型不是int32,也不是int64。

  • ValueError - Tensor的shape长度小于 indices 的shape的最后一个维度。

支持平台:

GPU CPU

样例:

>>> input_x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]), mindspore.float32)
>>> indices = Tensor(np.array([[0, 0], [0, 0]]), mindspore.int32)
>>> updates = Tensor(np.array([1.0, 2.2]), mindspore.float32)
>>> # Next, demonstrate the approximate operation process of this operator:
>>> # 1, indices[0] = [0, 0], indices[1] = [0, 0]
>>> # 2, And input_x[0, 0] = -0.1
>>> # 3, So input_x[indices] = [-0.1, -0.1]
>>> # 4, Satisfy the above formula: input_x[indices].shape=(2) == updates.shape=(2)
>>> # 5, Perform the multiply operation for the first time:
>>> #      first_input_x = input_x[0][0] * updates[0] = [[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]
>>> # 6, Perform the multiply operation for the second time:
>>> #      second_input_x = input_x[0][0] * updates[1] = [[-0.22, 0.3, 3.6], [0.4, 0.5, -3.2]]
>>> output = input_x.scatter_mul(indices, updates)
>>> print(output)
[[-0.22  0.3   3.6  ]
 [ 0.4   0.5   -3.2 ]]
scatter_sub(indices, updates)[源代码]

根据指定的更新值和输入索引,通过减法进行运算,将结果赋值到输出Tensor中。当同一索引有不同值时,更新的结果将分别减去这些值。此操作几乎等同于使用 mindspore.ops.ScatterNdSub ,只是更新后的结果是通过算子output返回,而不是直接原地更新input。

indices 的最后一个轴是每个索引向量的深度。对于每个索引向量, updates 中必须有相应的值。updates 的shape应该等于 input_x[indices] 的shape,其中 input_x 指当前Tensor。有关更多详细信息,请参见使用用例。

Note

GPU平台上,如果 indices 的某些值超出范围,则相应的 updates 不会更新到 input_x ,而不是抛出索引错误;CPU平台上直接抛出索引错误;Ascend平台不支持越界检查,若越界可能会造成未知错误。

参数:

  • indices (Tensor) - Tensor的索引,数据类型为int32或int64。其rank至少为2。

  • updates (Tensor) - 指定与本Tensor相减操作的Tensor,其数据类型与该Tensor相同。 updates.shape 应等于 indices.shape[:-1] + self.shape[indices.shape[-1]:]

返回:

Tensor,shape和数据类型与原Tensor相同。

异常:

  • TypeError - indices 的数据类型既不是int32,也不是int64。

  • ValueError - Tensor的shape长度小于 indices 的shape的最后一个维度。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[-0.1, 0.3, 3.6], [0.4, 0.5, -3.2]]).astype('float32'))
>>> indices = Tensor(np.array([[0, 0], [0, 0]]).astype('int32'))
>>> updates = Tensor(np.array([1.0, 2.2]).astype('float32'))
>>> output = x.scatter_sub(indices, updates)
>>> print(output)
[[-3.3000002  0.3        3.6      ]
[ 0.4        0.5       -3.2      ]]
split(axis=0, output_num=1)[源代码]

根据指定的轴和分割数量对Tensor进行分割。

Tensor将被分割为相同shape的子Tensor,且要求 self.shape(axis) 可被 output_num 整除。

参数:

  • axis (int) - 指定分割轴。默认值:0。

  • output_num (int) - 指定分割数量。其值为正整数。默认值:1。

返回:

tuple[Tensor],每个输出Tensor的shape相同,即 \((y_1, y_2, ..., y_S)\) 。数据类型与Tensor相同。

异常:

  • TypeError - axisoutput_num 不是int。

  • ValueError - axis 超出[-len(self.shape), len(self.shape))范围。或 output_num 小于或等于0。

  • ValueError - self.shape(axis) 不可被 output_num 整除。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([[1, 1, 1, 1], [2, 2, 2, 2]]), mindspore.int32)
>>> print(x)
[[1 1 1 1]
 [2 2 2 2]]
>>> output = x.split(1, 2)
>>> print(output)
(Tensor(shape=[2, 2], dtype=Int32, value=
[[1, 1],
 [2, 2]]), Tensor(shape=[2, 2], dtype=Int32, value=
[[1, 1],
 [2, 2]]))
to_coo()[源代码]

将常规Tensor转为稀疏化的COOTensor。

Note

现在只支持2维Tensor。

返回:

返回一个2维的COOTensor,是原稠密Tensor的稀疏化表示。其中数据分别为:

  • indices (Tensor) - 二维整数张量,其中N和ndims分别表示稀疏张量中 values 的数量和COOTensor维度的数量。

  • values (Tensor) - 一维张量,用来给 indices 中的每个元素提供数值。

  • shape (tuple(int)) - 整数元组,用来指定稀疏矩阵的稠密形状。目前只支持2维Tensor输入,所以 shape 长度只能为2。

异常:

  • ValueError - Tensor的shape不是2维。

支持平台:

GPU

样例:

>>> import numpy as np
>>> import mindspore
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1,  0], [-5, 0]]), mindspore.float32)
>>> output = x.to_coo()
>>> print(output.indices, output.values, output.shape)
[[0 0]
 [1 0]] [ 1. -5.] (2, 2)
to_csr()[源代码]

将常规Tensor转为稀疏化的CSRTensor。

Note

现在只支持2维Tensor。

返回:

返回一个2维的CSRTensor,是原稠密Tensor的稀疏化表示。其中数据分别为:

  • indptr (Tensor) - 一维整数张量,其中M等于 shape[0] + 1 , 表示每行非零元素的在 values 中存储的起止位置。

  • indices (Tensor) - 一维整数张量,其中N等于非零元素数量,表示每个元素的列索引值。

  • values (Tensor) - 一维张量,用来表示索引对应的数值。

  • shape (tuple(int)) - 整数元组,用来指定稀疏矩阵的稠密形状。目前只支持2维Tensor输入,所以 shape 长度只能为2。

异常:

  • ValueError - Tensor的shape不是2维。

支持平台:

GPU

样例:

>>> import numpy as np
>>> import mindspore
>>> from mindspore import Tensor
>>> x = Tensor(np.array([[1,  0], [-5, 0]]), mindspore.float32)
>>> output = x.to_csr()
>>> print(output.indptr, output.indices, output.values, output.shape)
[0 1 2] [0 0] [ 1. -5.] (2, 2)
to_tensor(slice_index=None, shape=None, opt_shard_group=None)[源代码]

返回init_data()的结果,并获取此Tensor的数据。

Note

不建议使用 to_tensor。请使用 init_data

参数:

  • slice_index (int) - 参数切片的索引。在初始化参数切片的时候使用,保证使用相同切片的设备可以生成相同的Tensor。默认值:None。

  • shape (list[int]) - 切片的shape,在初始化参数切片时使用。默认值:None。

  • opt_shard_group (str) - 优化器分片组,在自动或半自动并行模式下用于获取参数切片的分片。默认值:None。

返回:

初始化的Tensor。

异常:

  • TypeError - indices 的数据类型既不是int32,也不是int64。

  • ValueError - Tensor的shape长度小于 indices 的shape的最后一个维度。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore as ms
>>> from mindspore.common.initializer import initializer, Constant
>>> x = initializer(Constant(1), [2, 2], ms.float32)
>>> out = x.to_tensor()
>>> print(out)
[[1. 1.]
 [1. 1.]]
top_k(k, sorted=True)[源代码]

沿最后一个维度查找 k 个最大元素和对应的索引。

Warning

  • 如果 sorted 设置为’False’,它将使用aicpu运算符,性能可能会降低。

x 指的当前 Tensor。

如果 input_x 是一维Tensor,则查找Tensor中 k 个最大元素,并将其值和索引输出为Tensor。因此, values[k]input_xk 个最大元素,其索引是 indices[k]

对于多维矩阵,计算每行中最大的 k 个元素(沿最后一个维度的相应向量),因此:

\[values.shape = indices.shape = input\_x.shape[:-1] + [k].\]

如果两个比较的元素相同,则优先返回索引值较小的元素。

参数:

  • k (int) - 指定计算最大元素的数量,需要是常量。

  • sorted (bool, optional) - 如果为True,则获取的元素将按值降序排序。默认值:True。

返回:

2个Tensor组成的tuple, valuesindices

  • values (Tensor) - 最后一个维度的每个切片中的 k 最大元素。

  • indices (Tensor) - k 最大元素的对应索引。

异常:

  • TypeError - 如果 k 不是int。

  • TypeError - 如果 sorted 不是bool。

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore as ms
>>> from mindspore import Tensor
>>> input_x = Tensor([1, 2, 3, 4, 5], ms.float16)
>>> k = 3
>>> values, indices = input_x.top_k(k, sorted=True)
>>> print((values, indices))
(Tensor(shape=[3], dtype=Float16, value= [ 5.0000e+00,  4.0000e+00,  3.0000e+00]), Tensor(shape=[3],
  dtype=Int32, value= [4, 3, 2]))
trace(offset=0, axis1=0, axis2=1, dtype=None)[源代码]

在Tensor的对角线方向上的总和。

参数:

  • offset (int, optional) - 对角线与主对角线的偏移。可以是正值或负值。默认为主对角线。

  • axis1 (int, optional) - 二维子数组的第一轴,对角线应该从这里开始。默认为第一轴(0)。

  • axis2 (int, optional) - 二维子数组的第二轴,对角线应该从这里开始。默认为第二轴。

  • dtype (mindspore.dtype , optional) - 默认值为None。覆盖输出Tensor的dtype。

返回:

Tensor,对角线方向上的总和。

异常:

ValueError - 输入Tensor的维度少于2。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.eye(3, dtype=np.float32))
>>> print(x.trace())
3.0
unique_with_pad(pad_num)[源代码]

对当前一维张量中元素去重,返回一维张量中的唯一元素(使用pad_num填充)和相对索引。

基本操作与unique相同,但unique_with_pad多了pad操作。 unique运算符对张量处理后所返回的元组( yidx ), yidx 的shape通常会有差别,因此,为了解决上述情况, unique_with_pad操作符将用用户指定的 pad_num 填充 y 张量,使其具有与张量 idx 相同的形状。

参数:

  • pad_num (int) - 填充值。数据类型为int32或int64。

返回:

Tuple, (y, idx)y 是与当前张量形状和数据类型相同的Tensor,包含当前张量中去重后的元素,并用 pad_num 填充。 idx 为索引Tensor,包含当前张量中的元素在 y 中的索引,与当前张量的shape相同。

异常:

  • TypeError - 当前张量的数据类型既不是int32也不是int64。

  • ValueError - 当前张量不是一维张量。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore import dtype as mstype
>>> x = Tensor(np.array([1, 1, 2, 2, 3, 1]), mstype.int32)
>>> output, idx = x.unique_with_pad(pad_num=0)
>>> print(output)
[1 2 3 0 0 0]
>>> print(idx)
[0 0 1 1 2 0]
transpose(*axes)[源代码]

返回被转置后的Tensor。

  • 对于一维Tensor,这没有影响,因为转置后的向量是相同的。

  • 对于二维Tensor,是标准的矩阵转置。

  • 对于n维Tensor,如果提供了维度,则它们的顺序代表维度的置换方式。

如果未提供轴,且Tensor.shape等于(i[0], i[1],…i[n-2], i[n-1]),则Tensor.transpose().shape等于(i[n-1], i[n-2], … i[1], i[0])。

参数:

  • axes (Union[None, tuple(int), list(int), int], optional) - 如果 axes 为None或未设置,则该方法将反转维度。如果 axes 为tuple(int)或list(int),则Tensor.transpose()把Tensor转置为新的维度。如果 axes 为整数,则此表单仅作为元组/列表表单的备选。

返回:

Tensor,具有与输入Tensor相同的维度,其中维度被准确的排列。

异常:

  • TypeError - 输入参数类型有误。

  • ValueError - axes 的数量不等于Tensor.ndim。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> x = Tensor(np.ones((1,2,3), dtype=np.float32))
>>> x = x.transpose()
>>> print(x.shape)
(3, 2, 1)
unique_consecutive(return_idx=False, return_counts=False, axis=None)[源代码]

返回输入张量中每个连续等效元素组中唯一的元素。

参数:

  • return_idx (bool, optional) - 是否返回原始输入中,各元素在返回的唯一列表中的结束位置的索引。默认值:False。

  • return_counts (bool, optional) - 是否返回每个唯一元素的计数。默认值:False。

  • axis (int, optional) - 维度。如果为None,对输入进行展平操作,返回其唯一性。如果指定,必须是int32或int64类型。默认值:None。

返回:

Tensor或包含Tensor对象的元组( outputidxcounts )。 output 与输入张量具有相同的类型,用于表示唯一标量元素的输出列表。 如果 return_idx 为 True,则会有一个额外的返回张量 idx,它的形状与输入张量相同,表示原始输入中的元素映射到输出中的位置的索引。如果 return_idx 为 True,则会有一个额外的返回张量 counts,表示每个唯一值或张量的出现次数。

异常:

  • RuntimeError - axis 不在 [-ndim, ndim-1] 范围内。

支持平台:

Ascend GPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> from mindspore import dtype as mstype
>>> x = Tensor(np.array([1, 1, 2, 2, 3, 1, 1, 2]), mstype.int32)
>>> output, idx, counts = x.unique_consecutive(return_idx=True, return_counts=True, axis=None)
>>> print(output)
[1 2 3 1 2]
>>> print(idx)
[0 0 1 1 2 3 3 4]
>>> print(counts)
[2 2 1 2 1]
var(axis=None, ddof=0, keepdims=False)[源代码]

在指定维度上的方差。

方差是平均值的平方偏差的平均值,即:\(var = mean(abs(x - x.mean())**2)\)

返回方差值。默认情况下计算展开Tensor的方差,否则在指定维度上计算。

Note

不支持NumPy参数 dtypeoutwhere

参数:

  • axis (Union[None, int, tuple(int)]) - 维度,在指定维度上计算方差。其默认值是展开Tensor的方差。默认值:None。

  • ddof (int) - δ自由度。默认值:0。计算中使用的除数是 \(N - ddof\) ,其中 \(N\) 表示元素的数量。

  • keepdims (bool) - 默认值:False。

返回:

含有方差值的Tensor。

支持平台:

Ascend GPU CPU

样例:

>>> import numpy as np
>>> from mindspore import Tensor
>>> input_x = Tensor(np.array([1., 2., 3., 4.], np.float32))
>>> output = input_x.var()
>>> print(output)
1.25
view(*shape)[源代码]

根据输入shape重新创建一个Tensor,与原Tensor数据相同。该方法与reshape方法相同,都是依靠底层reshape算子实现的。

参数:

  • shape (Union[tuple(int), int]) - 输出Tensor的维度。

返回:

Tensor,具有与入参 shape 相同的维度。

样例:

>>> from mindspore import Tensor
>>> import numpy as np
>>> a = Tensor(np.array([[1, 2, 3], [2, 3, 4]], dtype=np.float32))
>>> output = a.view((3, 2))
>>> print(output)
[[1. 2.]
[3. 2.]
[3. 4.]]
xdivy(y)[源代码]

计算原Tensor除以输入的Tensor。当原Tensor为零时,则返回零。原Tensor的数据类型需要是float,complex或bool。 后面为了使表达清晰,使用 x 代替原Tensor。

\[out_i = x_{i}\y_{i}\]

xy 的输入遵循隐式类型转换规则使数据类型一致。y必须是一个Tensor或Scalar,当y是Tensor时,x和y的数据类型不能同时是bool类型,它们的shape可以广播。当y是Scalar时,只能是一个常量。

参数:

  • y (Union[Tensor, number.Number, bool]) - 当第一个输入x为Tensor的时候, 第二个输入y可以是Number类型、bool类型或者数据类型为float16、float32、float64、complex64、complex128、bool的Tensor。

返回:

Tensor,shape与广播后的shape相同,数据类型为两个输入中精度较高或数数值较高的类型。

异常:

  • TypeError - 如果 y 不是以下之一:Tensor、Number、bool。

  • TypeError - 如果 xy 的数据类型不是float16、float32、float64、complex64、complex128、bool。

  • ValueError - 如果 x 不能广播至 y 的shape。

  • RuntimeError - 如果Parameter的 x , y 需要进行数据类型转换,但是Parameter是不支持数据类型转换。

支持平台:

Ascend GPU CPU

样例:

>>> x = Tensor(np.array([2, 4, -1]), mindspore.float32)
>>> y = Tensor(np.array([2, 2, 2]), mindspore.float32)
>>> output = x.xdivy(y)
>>> print(output)
[ 1.   2.  -0.5]
xlogy(y)[源代码]

计算原Tensor乘以输入Tensor的对数。当原Tensor为零时,则返回零。原Tensor的数据类型需要是 numberbool_。 后面为了使表达清晰,使用 x 代替原Tensor。

\[out_i = x_{i}\ln{y_{i}}\]

xy 的输入遵循隐式类型转换规则,使数据类型一致。输入必须是两个Tensor或一个Tensor和一个Scalar。当输入是两个Tensor时,它们的数据类型不能同时是bool的,它们的shape可以广播。当输入是一个Tensor和一个Scalar时,Scalar只能是一个常量。

Warning

  • 在Ascend上, xy 必须为float16或float32。

参数:

  • y (Union[Tensor, number.Number, bool]) - 第二个输入为数值型。当第一个输入是Tensor或数据类型为数值型或bool的Tensor时,则第二个输入是数值型或bool。当第一个输入是Scalar时,则第二个输入必须是数据类型为数值型或bool的Tensor。

返回:

Tensor,shape与广播后的shape相同,数据类型为两个输入中精度较高或数数值较高的类型。

异常:

  • TypeError - 如果 xy 不是数值型、bool或Tensor。

  • TypeError - 如果 xy 的数据类型不是float16、float32、float64、complex64或complex128。

  • ValueError - 如果 x 不能广播到与 y 的shape一致。

支持平台:

Ascend CPU

样例:

>>> x = Tensor(np.array([-5, 0, 4]), mindspore.float32)
>>> y = Tensor(np.array([2, 2, 2]), mindspore.float32)
>>> print(x.xlogy(y))
[-3.465736   0.        2.7725887]