华为CANN/pypto全1张量创建函数

pypto.ones

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto

产品支持情况

功能说明

创建一个大小为size、填充值全为1的Tensor。其数据类型由dtype指定，默认数据类型为DT_FP32。

函数原型

ones(*size: Union[int, Sequence[int]], dtype: Optional[DataType] = None) -> Tensor

参数说明

返回值说明

返回输出Tensor，Tensor的数据类型由dtype决定，Shape为size大小，全部的值均为1。

约束说明

1. tileshape的维度需要与输出 result 维度相同，用于切分 result。

调用示例

TileShape设置示例

调用该operation接口前，应通过set_vec_tile_shapes设置TileShape。TileShape维度应和输出一致。 如输入size为[m, n]，输出为[m, n]，TileShape设置为[m1, n1]，则m1,n1分别用于切分m,n轴。

pypto.set_vec_tile_shapes(2, 3)

接口调用示例

# 示例1：使用可变参数传入size，使用默认dtype (DT_FP32) x1 = pypto.ones(2, 3) # 示例2：使用列表传入size，显式指定dtype (DT_INT32) x2 = pypto.ones([2, 3], dtype=pypto.DT_INT32)

结果示例如下：

x1输出数据: [[1., 1., 1.], [1., 1., 1.]] x2输出数据: [[1, 1, 1], [1, 1, 1]]

【免费下载链接】pyptoPyPTO（发音: pai p-t-o）：Parallel Tensor/Tile Operation编程范式。项目地址: https://gitcode.com/cann/pypto