mindspore.Tensor.short

查看源文件
mindspore.Tensor.short()[源代码]

将输入Tensor转换为 int16 类型并返回一个拷贝,与 self.astype(mstype.int16) 等价,其中Tensor中的值为浮点数时,则会丢弃小数部分,具体请参考 mindspore.Tensor.astype()

返回:

Tensor,其数据类型为 int16

支持平台:

Ascend GPU CPU

样例:

>>> import mindspore as ms
>>> import numpy as np
>>> x = ms.Tensor(np.array([1,2,3,4,5]), ms.int32)
>>> output = x.short()
>>> output
Tensor(shape=[5], dtype=Int16, value= [1, 2, 3, 4, 5])