CANN/pypto正态分布随机数生成函数文档

pypto.normal

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

产品支持情况

功能说明

生成指定shape的正态分布（高斯分布）随机数，其元素服从均值为0，方差为1。 $$ x_i \sim N(0, 1) $$

函数原型

normal(shape: List[int], key: List[int], counter: List[int], alg: List[int], dtype: DataType) -> Tensor

参数说明

约束说明

• 只在A5上支持。
• 不支持shape切分多个view shape，view shape必须和输入的shape一致。
• 不支持shape切分多个tile shape，tile shape必须和输入的shape一致。

返回值说明

返回一个指定shape、数据类型为dtype的Tensor，其元素服从均值为0，方差为1的正态分布。

调用示例

TileShape设置示例

调用该operation接口前，应通过set_vec_tile_shapes设置TileShape。

TileShape维度应和输入一致，且必须和输入的shape一致。

如输入shape为[m, n]，输出为[m, n]，TileShape设置为[m, n]。

pypto.set_vec_tile_shapes(4, 4)

接口调用示例

shape = [4, 4] key = [1234] counter = [0, 1] alg = [1] dtype = pypto.DT_FP32 y = pypto.normal(shape, key, counter, alg, dtype)

结果示例如下：

输出数据y: [[-0.32364845 1.8577391 0.39556974 0.2311697 ] [ 0.24243996 -1.9485782 -0.12983137 2.7137496 ] [ 1.6558666 2.0938187 -0.90338254 0.8765667 ] [ 0.86518306 0.01034508 0.2893259 0.01748212]]

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