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