CANN/GE Tensor张量类API文档

CANN/GE Tensor张量类API文档 Tensor【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge产品支持情况产品是否支持Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√模块导入from ge.graph import Tensor功能说明Tensor 类是张量数据类支持通过内存数据或文件创建张量。支持设置和获取张量的格式Format、数据类型DataType、形状Shape、数据TensorLike以及放置位置Placement。张量支持在 Host 和 Device 之间迁移。函数原型构造函数Tensor(dataNone, file_pathNone, data_typeDataType.DT_FLOAT, formatFormat.FORMAT_ND, shapeNone, placementPlacement.PLACEMENT_HOST)属性property format - Formatproperty data_type - DataTypeproperty shape - Shapeproperty data - TensorLikeproperty placement - Placement方法set_format(format: Format) - Tensorget_format() - Formatset_data_type(data_type: DataType) - Tensorget_data_type() - DataTypeget_shape() - Shapeget_data() - TensorLikeget_tensor_desc() - TensorDescget_placement() - Placementto_host() - Tensorto_device() - Tensor参数说明构造函数参数参数名类型是否必选说明dataUnion[List[int], List[float], List[bool], None]否内存数据通过列表传入张量数据。与 file_path 二选一不能同时指定。file_pathUnion[str, None]否文件路径从文件读取张量数据。与 data 二选一不能同时指定。data_typeDataType否张量的数据类型使用 DataType 枚举值默认值为 DataType.DT_FLOAT。formatFormat否张量的数据格式使用 Format 枚举值默认值为 Format.FORMAT_ND。shapeUnion[List[int], None]否张量的形状用整数列表表示各维度大小。若为 None表示标量。placementPlacement否张量的放置位置使用 Placement 枚举值默认值为 Placement.PLACEMENT_HOST。set_format 参数参数名类型是否必选说明formatFormat是目标数据格式使用 Format 枚举值。set_data_type 参数参数名类型是否必选说明data_typeDataType是目标数据类型使用 DataType 枚举值。返回值说明构造函数返回 Tensor 对象实例。属性返回值属性返回类型说明formatFormat张量的数据格式。data_typeDataType张量的数据类型。shapeShape张量的形状信息。dataTensorLike张量的数据内容。标量返回单个数值非标量返回嵌套列表结构。placementPlacement张量的放置位置。方法返回值方法返回类型说明set_formatTensor返回自身支持链式调用。get_formatFormat返回张量的数据格式。set_data_typeTensor返回自身支持链式调用。get_data_typeDataType返回张量的数据类型。get_shapeShape返回张量的形状信息。get_dataTensorLike返回张量的数据内容。标量返回单个数值非标量返回嵌套列表结构。get_tensor_descTensorDesc返回张量的描述信息TensorDesc 对象。get_placementPlacement返回张量的放置位置。to_hostTensor返回自身将张量从 Device 迁移到 Host。to_deviceTensor返回自身将张量从 Host 迁移到 Device。约束说明构造张量时data 和 file_path 只能指定其中一个不能同时指定也不能都不指定都不指定时创建空张量。支持的数据类型包括DT_FLOAT、DT_FLOAT16、DT_INT8、DT_INT32、DT_UINT8、DT_INT16、DT_UINT16、DT_UINT32、DT_INT64、DT_UINT64、DT_BOOL。不支持 DT_DOUBLE。shape 参数必须为整数列表list of int若为 None 则表示标量。placement 参数必须为 Placement 枚举值。Tensor 不支持拷贝copy和深拷贝deepcopy。to_host() 仅适用于当前位于 Device 的张量to_device() 仅适用于当前位于 Host 的张量。【免费下载链接】geGEGraph Engine是面向昇腾的图编译器和执行器提供了计算图优化、多流并行、内存复用和模型下沉等技术手段加速模型执行效率减少模型内存占用。 GE 提供对 PyTorch、TensorFlow 前端的友好接入能力并同时支持 onnx、pb 等主流模型格式的解析与编译。项目地址: https://gitcode.com/cann/ge创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考