mindspore.tensor

查看源文件
mindspore.tensor(input_data=None, dtype=None, shape=None, init=None, internal=False, const_arg=False)[源代码]

此接口用于在Cell.construct()或者@jit装饰的函数内,创建一个新的Tensor对象。

在图模式下,MindSpore可以在运行时依据 dtype 参数来动态创建新Tensor。

详情请参考教程 创建和使用Tensor

有别于Tensor类,其与Tensor类的区别为内部增加了 Annotation 指示当前创建的Tensor的类型,与Tensor类相比能够防止AnyType的产生。

参数和返回值与Tensor类完全一致。另参考:mindspore.Tensor

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore as ms
>>> from mindspore import jit, tensor
>>> @jit
... def func(x):
...    return tensor(x.asnumpy(), dtype=ms.float32)
>>> x = tensor([1, 2, 3])
>>> y = func(x)
>>> print(y)
[1. 2. 3.]