geometric_

paddle. geometric_ ( x, probs, name=None ) [源代码]

使用从几何分布中抽取的数据填充 Tensor。

参数

  • x (Tensor) - 被填充的 Tensor ，数据类型为： float32、float64。

  • probs (float | Tensor) - 输入的概率参数， probs 的值必须为正，数据类型为 float、Tensor。当该参数类型为 Tensor 时， probs 代表每次试验成功的概率。

  • name (str ，可选) - 具体用法请参见 Name，一般无需设置，默认值为 None。

返回

填充了服从几何分布随机数的输入 Tensor。

代码示例

>>> import paddle
>>> x = paddle.randn([3, 4])
>>> x.geometric_(0.3)
>>> print(x)
Tensor(shape=[3, 4], dtype=float32, place=Place(cpu), stop_gradient=True,
[[2.42739224, 4.78268528, 1.23302543, 3.76555204],
 [1.38877118, 0.16075331, 0.16401523, 2.47349310],
 [1.72872102, 2.76533413, 0.33410925, 1.63351011]])