CANN/pypto：Tensor构造函数

pypto.Tensor构造函数【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto产品支持情况产品是否支持Atlas A3 训练系列产品/Atlas A3 推理系列产品√Atlas A2 训练系列产品/Atlas A2 推理系列产品√功能说明创建Tensor对象。Tensor创建时为未初始化的随机值。函数原型__init__(self, shapeNone, dtype: Union[DataType, None] None, name: str , format: TileOpFormat TileOpFormat.TILEOP_ND, data_ptr: Optional[int] None, deviceNone, ori_shapeNone )参数说明参数名输入/输出说明shape输入Tensor的形状可以是以下类型- None创建空Tensor- List[int]整数列表指定各维度的大小- List[Union[int, SymbolicScalar]]包含整数或符号标量的列表用于动态形状dtype输入Tensor的数据类型。name输入Tensor的名称。format输入Tensor的格式可选值包括- TileOpFormat.TILEOP_ND(默认)- TileOpFormat.TILEOP_NZdata_ptr输入数据指针默认为None, 当前仅前端框架内部使用算子开发人员可忽略device输入设备信息默认为Noneori_shape输入原始形状用于保存Tensor的原始形状信息默认为None返回值说明Tensor对象。约束说明无。调用示例# 创建空Tensor empty_tensor pypto.Tensor() # 创建指定形状和数据类型的Tensor tensor1 pypto.Tensor(shape(4, 4), dtypepypto.DT_FP32) tensor2 pypto.Tensor(shape[8, 16, 32], dtypepypto.DT_INT32) # 创建带名称的Tensor named_tensor pypto.Tensor(shape(4, 4), dtypepypto.DT_FP32, nameinput_tensor ) # 创建指定格式的Tensor sparse_tensor pypto.Tensor(shape(4, 32), dtypepypto.DT_FP32, formatpypto.TileOpFormat.TILEOP_NZ ) # 创建动态Shape的Tensor使用符号化标量 dynamic_shape [pypto.SymbolicScalar(N), 4, 8] dynamic_tensor pypto.Tensor(shapedynamic_shape, dtypepypto.DT_FP32 ) # 使用 pypto.tensor 便捷函数创建推荐方式 tensor3 pypto.tensor((4, 4), pypto.DT_FP32) tensor4 pypto.tensor((4, 4), pypto.DT_FP32, namemy_tensor)【免费下载链接】pyptoPyPTO发音: pai p-t-oParallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考