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