numpy

产品支持情况

产品

是否支持

Atlas 800I A2 推理产品/A200I A2 Box 异构组件

Atlas A2 训练系列产品

x

Atlas 200I/500 A2 推理产品

x

Atlas 推理系列产品

x

Atlas 训练系列产品

x

函数功能

获取tensor的numpy数据。

函数原型

1
numpy(copy=False)

参数说明

参数名称

数据类型

取值说明

copy

bool

取值为False或者True。

默认值为False,表示从tensor转换到numpy.ndarray,且数据不做拷贝,如果取值为True,则表示需要对数据进行拷贝。

如果是tensor是string类型的数据,该参数需要用户设置成True,否则会抛出异常。

调用示例

1
2
tensor = Tensor(numpy.array([1]))
np_arr = tensor.numpy()

返回值

返回numpy.array。

约束说明