pypto.normal

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√

功能说明

生成指定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

参数说明

参数名	输入/输出	说明
shape	输入	输出Tensor的形状。 长度支持1-4维。
key	输入	随机数生成器的seed。 长度仅支持为1。
counter	输入	随机数生成器的计数器。 长度仅支持为2。
alg	输入	随机数生成算法，当前仅支持值1（Philox算法），3（auto_select，选择Philox算法）。 长度仅支持为1。
dtype	输入	输出Tensor的数据类型。 支持的数据类型为：DT_FP32, DT_FP16, DT_BF16。

约束说明

• 只在A5上支持。

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

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

• tile shape尾轴必须是4的倍数。

返回值说明

返回一个指定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]]