Skip to content

P
Paddle

PaddlePaddle / Paddle
大约 2 年 前同步成功

通知 2325
Star 20933
Fork 5424
P
Paddle
未验证 提交 84fe0454 编写于 作者: G gouzil 提交者： GitHub
浏览文件
操作

[xdoctest] No.44-47 and No.50-59 doc style (#55813)

上级 30a02d27
展开全部 隐藏空白更改
内联 并排
Showing with 538 addition and 519 deletion
python/paddle/distribution/cauchy.py
浏览文件 @ 84fe0454
...@@ -39,20 +39,20 @@ class Cauchy(distribution.Distribution): ...@@ -39,20 +39,20 @@ class Cauchy(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import Cauchy >>> from paddle.distribution import Cauchy
# init Cauchy with float >>> # init Cauchy with float
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.entropy()) >>> print(rv.entropy())
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# 2.71334577) 2.71334577)
# init Cauchy with N-Dim tensor >>> # init Cauchy with N-Dim tensor
rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.entropy()) >>> print(rv.entropy())
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [2.53102422, 3.22417140]) [2.53102422, 3.22417140])
""" """
def __init__(self, loc, scale, name=None): def __init__(self, loc, scale, name=None):
...@@ -114,32 +114,32 @@ class Cauchy(distribution.Distribution): ...@@ -114,32 +114,32 @@ class Cauchy(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import Cauchy >>> from paddle.distribution import Cauchy
# init Cauchy with float >>> # init Cauchy with float
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.sample([10]).shape) >>> print(rv.sample([10]).shape)
# [10] [10]
# init Cauchy with 0-Dim tensor >>> # init Cauchy with 0-Dim tensor
rv = Cauchy(loc=paddle.full((), 0.1), scale=paddle.full((), 1.2)) >>> rv = Cauchy(loc=paddle.full((), 0.1), scale=paddle.full((), 1.2))
print(rv.sample([10]).shape) >>> print(rv.sample([10]).shape)
# [10] [10]
# init Cauchy with N-Dim tensor >>> # init Cauchy with N-Dim tensor
rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.sample([10]).shape) >>> print(rv.sample([10]).shape)
# [10, 2] [10, 2]
# sample 2-Dim data >>> # sample 2-Dim data
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.sample([10, 2]).shape) >>> print(rv.sample([10, 2]).shape)
# [10, 2] [10, 2]
rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.sample([10, 2]).shape) >>> print(rv.sample([10, 2]).shape)
# [10, 2, 2] [10, 2, 2]
""" """
name = name if name is not None else (self.name + '_sample') name = name if name is not None else (self.name + '_sample')
with paddle.no_grad(): with paddle.no_grad():
...@@ -159,32 +159,32 @@ class Cauchy(distribution.Distribution): ...@@ -159,32 +159,32 @@ class Cauchy(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import Cauchy >>> from paddle.distribution import Cauchy
# init Cauchy with float >>> # init Cauchy with float
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.rsample([10]).shape) >>> print(rv.rsample([10]).shape)
# [10] [10]
# init Cauchy with 0-Dim tensor >>> # init Cauchy with 0-Dim tensor
rv = Cauchy(loc=paddle.full((), 0.1), scale=paddle.full((), 1.2)) >>> rv = Cauchy(loc=paddle.full((), 0.1), scale=paddle.full((), 1.2))
print(rv.rsample([10]).shape) >>> print(rv.rsample([10]).shape)
# [10] [10]
# init Cauchy with N-Dim tensor >>> # init Cauchy with N-Dim tensor
rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.rsample([10]).shape) >>> print(rv.rsample([10]).shape)
# [10, 2] [10, 2]
# sample 2-Dim data >>> # sample 2-Dim data
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.rsample([10, 2]).shape) >>> print(rv.rsample([10, 2]).shape)
# [10, 2] [10, 2]
rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.rsample([10, 2]).shape) >>> print(rv.rsample([10, 2]).shape)
# [10, 2, 2] [10, 2, 2]
""" """
name = name if name is not None else (self.name + '_rsample') name = name if name is not None else (self.name + '_rsample')
...@@ -222,32 +222,32 @@ class Cauchy(distribution.Distribution): ...@@ -222,32 +222,32 @@ class Cauchy(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import Cauchy >>> from paddle.distribution import Cauchy
# init Cauchy with float >>> # init Cauchy with float
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.prob(paddle.to_tensor(1.5))) >>> print(rv.prob(paddle.to_tensor(1.5)))
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# 0.11234467) 0.11234467)
# broadcast to value >>> # broadcast to value
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.prob(paddle.to_tensor([1.5, 5.1]))) >>> print(rv.prob(paddle.to_tensor([1.5, 5.1])))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [0.11234467, 0.01444674]) [0.11234467, 0.01444674])
# init Cauchy with N-Dim tensor >>> # init Cauchy with N-Dim tensor
rv = Cauchy(loc=paddle.to_tensor([0.1, 0.1]), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor([0.1, 0.1]), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.prob(paddle.to_tensor([1.5, 5.1]))) >>> print(rv.prob(paddle.to_tensor([1.5, 5.1])))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [0.10753712, 0.02195240]) [0.10753712, 0.02195240])
# init Cauchy with N-Dim tensor with broadcast >>> # init Cauchy with N-Dim tensor with broadcast
rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.prob(paddle.to_tensor([1.5, 5.1]))) >>> print(rv.prob(paddle.to_tensor([1.5, 5.1])))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [0.10753712, 0.02195240]) [0.10753712, 0.02195240])
""" """
name = self.name + '_prob' name = self.name + '_prob'
...@@ -271,32 +271,32 @@ class Cauchy(distribution.Distribution): ...@@ -271,32 +271,32 @@ class Cauchy(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import Cauchy >>> from paddle.distribution import Cauchy
# init Cauchy with float >>> # init Cauchy with float
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.log_prob(paddle.to_tensor(1.5))) >>> print(rv.log_prob(paddle.to_tensor(1.5)))
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# -2.18618369) -2.18618369)
# broadcast to value >>> # broadcast to value
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.log_prob(paddle.to_tensor([1.5, 5.1]))) >>> print(rv.log_prob(paddle.to_tensor([1.5, 5.1])))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [-2.18618369, -4.23728657]) [-2.18618369, -4.23728657])
# init Cauchy with N-Dim tensor >>> # init Cauchy with N-Dim tensor
rv = Cauchy(loc=paddle.to_tensor([0.1, 0.1]), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor([0.1, 0.1]), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.log_prob(paddle.to_tensor([1.5, 5.1]))) >>> print(rv.log_prob(paddle.to_tensor([1.5, 5.1])))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [-2.22991920, -3.81887865]) [-2.22991920, -3.81887865])
# init Cauchy with N-Dim tensor with broadcast >>> # init Cauchy with N-Dim tensor with broadcast
rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.log_prob(paddle.to_tensor([1.5, 5.1]))) >>> print(rv.log_prob(paddle.to_tensor([1.5, 5.1])))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [-2.22991920, -3.81887865]) [-2.22991920, -3.81887865])
""" """
name = self.name + '_log_prob' name = self.name + '_log_prob'
...@@ -338,32 +338,32 @@ class Cauchy(distribution.Distribution): ...@@ -338,32 +338,32 @@ class Cauchy(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import Cauchy >>> from paddle.distribution import Cauchy
# init Cauchy with float >>> # init Cauchy with float
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.cdf(paddle.to_tensor(1.5))) >>> print(rv.cdf(paddle.to_tensor(1.5)))
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# 0.77443725) 0.77443725)
# broadcast to value >>> # broadcast to value
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.cdf(paddle.to_tensor([1.5, 5.1]))) >>> print(rv.cdf(paddle.to_tensor([1.5, 5.1])))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [0.77443725, 0.92502367]) [0.77443725, 0.92502367])
# init Cauchy with N-Dim tensor >>> # init Cauchy with N-Dim tensor
rv = Cauchy(loc=paddle.to_tensor([0.1, 0.1]), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor([0.1, 0.1]), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.cdf(paddle.to_tensor([1.5, 5.1]))) >>> print(rv.cdf(paddle.to_tensor([1.5, 5.1])))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [0.80256844, 0.87888104]) [0.80256844, 0.87888104])
# init Cauchy with N-Dim tensor with broadcast >>> # init Cauchy with N-Dim tensor with broadcast
rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.cdf(paddle.to_tensor([1.5, 5.1]))) >>> print(rv.cdf(paddle.to_tensor([1.5, 5.1])))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [0.80256844, 0.87888104]) [0.80256844, 0.87888104])
""" """
name = self.name + '_cdf' name = self.name + '_cdf'
...@@ -399,20 +399,20 @@ class Cauchy(distribution.Distribution): ...@@ -399,20 +399,20 @@ class Cauchy(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import Cauchy >>> from paddle.distribution import Cauchy
# init Cauchy with float >>> # init Cauchy with float
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
print(rv.entropy()) >>> print(rv.entropy())
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# 2.71334577) 2.71334577)
# init Cauchy with N-Dim tensor >>> # init Cauchy with N-Dim tensor
rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0])) >>> rv = Cauchy(loc=paddle.to_tensor(0.1), scale=paddle.to_tensor([1.0, 2.0]))
print(rv.entropy()) >>> print(rv.entropy())
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [2.53102422, 3.22417140]) [2.53102422, 3.22417140])
""" """
name = self.name + '_entropy' name = self.name + '_entropy'
...@@ -438,14 +438,14 @@ class Cauchy(distribution.Distribution): ...@@ -438,14 +438,14 @@ class Cauchy(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import Cauchy >>> from paddle.distribution import Cauchy
rv = Cauchy(loc=0.1, scale=1.2) >>> rv = Cauchy(loc=0.1, scale=1.2)
rv_other = Cauchy(loc=paddle.to_tensor(1.2), scale=paddle.to_tensor([2.3, 3.4])) >>> rv_other = Cauchy(loc=paddle.to_tensor(1.2), scale=paddle.to_tensor([2.3, 3.4]))
print(rv.kl_divergence(rv_other)) >>> print(rv.kl_divergence(rv_other))
# Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(cpu), stop_gradient=True,
# [0.19819736, 0.31532931]) [0.19819736, 0.31532931])
""" """
name = self.name + '_kl_divergence' name = self.name + '_kl_divergence'
......
python/paddle/distribution/independent.py
浏览文件 @ 84fe0454
...@@ -31,21 +31,21 @@ class Independent(distribution.Distribution): ...@@ -31,21 +31,21 @@ class Independent(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import independent >>> from paddle.distribution import independent
beta = paddle.distribution.Beta(paddle.to_tensor([0.5, 0.5]), paddle.to_tensor([0.5, 0.5])) >>> beta = paddle.distribution.Beta(paddle.to_tensor([0.5, 0.5]), paddle.to_tensor([0.5, 0.5]))
print(beta.batch_shape, beta.event_shape) >>> print(beta.batch_shape, beta.event_shape)
# (2,) () (2,) ()
print(beta.log_prob(paddle.to_tensor(0.2))) >>> print(beta.log_prob(paddle.to_tensor(0.2)))
# Tensor(shape=[2], dtype=float32, place=Place(gpu:0), stop_gradient=True, Tensor(shape=[2], dtype=float32, place=Place(gpu:0), stop_gradient=True,
# [-0.22843921, -0.22843921]) [-0.22843921, -0.22843921])
reinterpreted_beta = independent.Independent(beta, 1) >>> reinterpreted_beta = independent.Independent(beta, 1)
print(reinterpreted_beta.batch_shape, reinterpreted_beta.event_shape) >>> print(reinterpreted_beta.batch_shape, reinterpreted_beta.event_shape)
# () (2,) () (2,)
print(reinterpreted_beta.log_prob(paddle.to_tensor([0.2, 0.2]))) >>> print(reinterpreted_beta.log_prob(paddle.to_tensor([0.2, 0.2])))
# Tensor(shape=[], dtype=float32, place=Place(gpu:0), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(gpu:0), stop_gradient=True,
# -0.45687842) -0.45687842)
""" """
def __init__(self, base, reinterpreted_batch_rank): def __init__(self, base, reinterpreted_batch_rank):
......
python/paddle/distribution/kl.py
浏览文件 @ 84fe0454
...@@ -53,14 +53,14 @@ def kl_divergence(p, q): ...@@ -53,14 +53,14 @@ def kl_divergence(p, q):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
p = paddle.distribution.Beta(alpha=0.5, beta=0.5) >>> p = paddle.distribution.Beta(alpha=0.5, beta=0.5)
q = paddle.distribution.Beta(alpha=0.3, beta=0.7) >>> q = paddle.distribution.Beta(alpha=0.3, beta=0.7)
print(paddle.distribution.kl_divergence(p, q)) >>> print(paddle.distribution.kl_divergence(p, q))
# Tensor(shape=[], dtype=float32, place=CUDAPlace(0), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# 0.21193528) 0.21193528)
""" """
return _dispatch(type(p), type(q))(p, q) return _dispatch(type(p), type(q))(p, q)
...@@ -82,11 +82,11 @@ def register_kl(cls_p, cls_q): ...@@ -82,11 +82,11 @@ def register_kl(cls_p, cls_q):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
@paddle.distribution.register_kl(paddle.distribution.Beta, paddle.distribution.Beta) >>> @paddle.distribution.register_kl(paddle.distribution.Beta, paddle.distribution.Beta)
def kl_beta_beta(): >>> def kl_beta_beta():
pass # insert implementation here ... pass # insert implementation here
""" """
if not issubclass(cls_p, Distribution) or not issubclass( if not issubclass(cls_p, Distribution) or not issubclass(
cls_q, Distribution cls_q, Distribution
......
python/paddle/distribution/laplace.py
浏览文件 @ 84fe0454
...@@ -44,12 +44,12 @@ class Laplace(distribution.Distribution): ...@@ -44,12 +44,12 @@ class Laplace(distribution.Distribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
>>> paddle.seed(2023)
m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0)) >>> m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0))
m.sample() # Laplace distributed with loc=0, scale=1 >>> m.sample() # Laplace distributed with loc=0, scale=1
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# 3.68546247) 1.31554604)
""" """
...@@ -173,13 +173,13 @@ class Laplace(distribution.Distribution): ...@@ -173,13 +173,13 @@ class Laplace(distribution.Distribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0)) >>> m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0))
value = paddle.to_tensor(0.1) >>> value = paddle.to_tensor(0.1)
m.log_prob(value) >>> m.log_prob(value)
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# -0.79314721) -0.79314721)
""" """
loc, scale, value = self._validate_value(value) loc, scale, value = self._validate_value(value)
...@@ -205,12 +205,12 @@ class Laplace(distribution.Distribution): ...@@ -205,12 +205,12 @@ class Laplace(distribution.Distribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0)) >>> m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0))
m.entropy() >>> m.entropy()
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# 1.69314718) 1.69314718)
""" """
return 1 + paddle.log(2 * self.scale) return 1 + paddle.log(2 * self.scale)
...@@ -236,13 +236,13 @@ class Laplace(distribution.Distribution): ...@@ -236,13 +236,13 @@ class Laplace(distribution.Distribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0)) >>> m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0))
value = paddle.to_tensor(0.1) >>> value = paddle.to_tensor(0.1)
m.cdf(value) >>> m.cdf(value)
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# 0.54758132) 0.54758132)
""" """
loc, scale, value = self._validate_value(value) loc, scale, value = self._validate_value(value)
iterm = ( iterm = (
...@@ -275,13 +275,12 @@ class Laplace(distribution.Distribution): ...@@ -275,13 +275,12 @@ class Laplace(distribution.Distribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
>>> m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0))
m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0)) >>> value = paddle.to_tensor(0.1)
value = paddle.to_tensor(0.1) >>> m.icdf(value)
m.icdf(value) Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, -1.60943794)
# -1.60943794)
""" """
loc, scale, value = self._validate_value(value) loc, scale, value = self._validate_value(value)
term = value - 0.5 term = value - 0.5
...@@ -300,12 +299,12 @@ class Laplace(distribution.Distribution): ...@@ -300,12 +299,12 @@ class Laplace(distribution.Distribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
>>> paddle.seed(2023)
m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0)) >>> m = paddle.distribution.Laplace(paddle.to_tensor(0.0), paddle.to_tensor(1.0))
m.sample() # Laplace distributed with loc=0, scale=1 >>> m.sample() # Laplace distributed with loc=0, scale=1
# Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
# 3.68546247) 1.31554604)
""" """
shape = shape if isinstance(shape, tuple) else tuple(shape) shape = shape if isinstance(shape, tuple) else tuple(shape)
with paddle.no_grad(): with paddle.no_grad():
...@@ -323,12 +322,12 @@ class Laplace(distribution.Distribution): ...@@ -323,12 +322,12 @@ class Laplace(distribution.Distribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
>>> paddle.seed(2023)
m = paddle.distribution.Laplace(paddle.to_tensor([0.0]), paddle.to_tensor([1.0])) >>> m = paddle.distribution.Laplace(paddle.to_tensor([0.0]), paddle.to_tensor([1.0]))
m.rsample((1,)) # Laplace distributed with loc=0, scale=1 >>> m.rsample((1,)) # Laplace distributed with loc=0, scale=1
# Tensor(shape=[1, 1], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[1, 1], dtype=float32, place=Place(cpu), stop_gradient=True,
# [[0.04337667]]) [[1.31554604]])
""" """
eps = self._get_eps() eps = self._get_eps()
...@@ -395,13 +394,13 @@ class Laplace(distribution.Distribution): ...@@ -395,13 +394,13 @@ class Laplace(distribution.Distribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
m1 = paddle.distribution.Laplace(paddle.to_tensor([0.0]), paddle.to_tensor([1.0])) >>> m1 = paddle.distribution.Laplace(paddle.to_tensor([0.0]), paddle.to_tensor([1.0]))
m2 = paddle.distribution.Laplace(paddle.to_tensor([1.0]), paddle.to_tensor([0.5])) >>> m2 = paddle.distribution.Laplace(paddle.to_tensor([1.0]), paddle.to_tensor([0.5]))
m1.kl_divergence(m2) >>> m1.kl_divergence(m2)
# Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True, Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True,
# [1.04261160]) [1.04261160])
""" """
var_ratio = other.scale / self.scale var_ratio = other.scale / self.scale
......
python/paddle/distribution/lognormal.py
浏览文件 @ 84fe0454
...@@ -49,36 +49,44 @@ class LogNormal(TransformedDistribution): ...@@ -49,36 +49,44 @@ class LogNormal(TransformedDistribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import LogNormal >>> from paddle.distribution import LogNormal
# Define a single scalar LogNormal distribution. >>> # Define a single scalar LogNormal distribution.
dist = LogNormal(loc=0., scale=3.) >>> dist = LogNormal(loc=0., scale=3.)
# Define a batch of two scalar valued LogNormals. >>> # Define a batch of two scalar valued LogNormals.
# The underlying Normal of first has mean 1 and standard deviation 11, the underlying Normal of second 2 and 22. >>> # The underlying Normal of first has mean 1 and standard deviation 11, the underlying Normal of second 2 and 22.
dist = LogNormal(loc=[1., 2.], scale=[11., 22.]) >>> dist = LogNormal(loc=[1., 2.], scale=[11., 22.])
# Get 3 samples, returning a 3 x 2 tensor. >>> # Get 3 samples, returning a 3 x 2 tensor.
dist.sample((3, )) >>> dist.sample((3, ))
# Define a batch of two scalar valued LogNormals. >>> # Define a batch of two scalar valued LogNormals.
# Their underlying Normal have mean 1, but different standard deviations. >>> # Their underlying Normal have mean 1, but different standard deviations.
dist = LogNormal(loc=1., scale=[11., 22.]) >>> dist = LogNormal(loc=1., scale=[11., 22.])
# Complete example >>> # Complete example
value_tensor = paddle.to_tensor([0.8], dtype="float32") >>> value_tensor = paddle.to_tensor([0.8], dtype="float32")
lognormal_a = LogNormal([0.], [1.]) >>> lognormal_a = LogNormal([0.], [1.])
lognormal_b = LogNormal([0.5], [2.]) >>> lognormal_b = LogNormal([0.5], [2.])
sample = lognormal_a.sample((2, )) >>> sample = lognormal_a.sample((2, ))
# a random tensor created by lognormal distribution with shape: [2, 1] >>> # a random tensor created by lognormal distribution with shape: [2, 1]
entropy = lognormal_a.entropy() >>> entropy = lognormal_a.entropy()
# [1.4189385] with shape: [1] >>> print(entropy)
lp = lognormal_a.log_prob(value_tensor) Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True,
# [-0.72069150] with shape: [1] [1.41893852])
p = lognormal_a.probs(value_tensor) >>> lp = lognormal_a.log_prob(value_tensor)
# [0.48641577] with shape: [1] >>> print(lp)
kl = lognormal_a.kl_divergence(lognormal_b) Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True,
# [0.34939718] with shape: [1] [-0.72069150])
>>> p = lognormal_a.probs(value_tensor)
>>> print(p)
Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True,
[0.48641577])
>>> kl = lognormal_a.kl_divergence(lognormal_b)
>>> print(kl)
Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True,
[0.34939718])
""" """
def __init__(self, loc, scale): def __init__(self, loc, scale):
......
python/paddle/distribution/multinomial.py
浏览文件 @ 84fe0454
...@@ -53,18 +53,17 @@ class Multinomial(distribution.Distribution): ...@@ -53,18 +53,17 @@ class Multinomial(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
>>> paddle.seed(2023)
multinomial = paddle.distribution.Multinomial(10, paddle.to_tensor([0.2, 0.3, 0.5])) >>> multinomial = paddle.distribution.Multinomial(10, paddle.to_tensor([0.2, 0.3, 0.5]))
print(multinomial.sample((2, 3))) >>> print(multinomial.sample((2, 3)))
# Tensor(shape=[2, 3, 3], dtype=float32, place=Place(gpu:0), stop_gradient=True, Tensor(shape=[2, 3, 3], dtype=float32, place=Place(cpu), stop_gradient=True,
# [[[1., 4., 5.], [[[1., 5., 4.],
# [0., 2., 8.], [0., 4., 6.],
# [2., 4., 4.]], [1., 3., 6.]],
[[2., 2., 6.],
# [[1., 6., 3.], [0., 6., 4.],
# [3., 3., 4.], [3., 3., 4.]]])
# [3., 4., 3.]]])
""" """
def __init__(self, total_count, probs): def __init__(self, total_count, probs):
......
python/paddle/distribution/normal.py
浏览文件 @ 84fe0454
...@@ -54,36 +54,44 @@ class Normal(distribution.Distribution): ...@@ -54,36 +54,44 @@ class Normal(distribution.Distribution):
Examples: Examples:
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import Normal >>> from paddle.distribution import Normal
# Define a single scalar Normal distribution. >>> # Define a single scalar Normal distribution.
dist = Normal(loc=0., scale=3.) >>> dist = Normal(loc=0., scale=3.)
# Define a batch of two scalar valued Normals. >>> # Define a batch of two scalar valued Normals.
# The first has mean 1 and standard deviation 11, the second 2 and 22. >>> # The first has mean 1 and standard deviation 11, the second 2 and 22.
dist = Normal(loc=[1., 2.], scale=[11., 22.]) >>> dist = Normal(loc=[1., 2.], scale=[11., 22.])
# Get 3 samples, returning a 3 x 2 tensor. >>> # Get 3 samples, returning a 3 x 2 tensor.
dist.sample([3]) >>> dist.sample([3])
# Define a batch of two scalar valued Normals. >>> # Define a batch of two scalar valued Normals.
# Both have mean 1, but different standard deviations. >>> # Both have mean 1, but different standard deviations.
dist = Normal(loc=1., scale=[11., 22.]) >>> dist = Normal(loc=1., scale=[11., 22.])
# Complete example >>> # Complete example
value_tensor = paddle.to_tensor([0.8], dtype="float32") >>> value_tensor = paddle.to_tensor([0.8], dtype="float32")
normal_a = Normal([0.], [1.]) >>> normal_a = Normal([0.], [1.])
normal_b = Normal([0.5], [2.]) >>> normal_b = Normal([0.5], [2.])
sample = normal_a.sample([2]) >>> sample = normal_a.sample([2])
# a random tensor created by normal distribution with shape: [2, 1] >>> # a random tensor created by normal distribution with shape: [2, 1]
entropy = normal_a.entropy() >>> entropy = normal_a.entropy()
# [1.4189385] with shape: [1] >>> print(entropy)
lp = normal_a.log_prob(value_tensor) Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True,
# [-1.2389386] with shape: [1] [1.41893852])
p = normal_a.probs(value_tensor) >>> lp = normal_a.log_prob(value_tensor)
# [0.28969154] with shape: [1] >>> print(lp)
kl = normal_a.kl_divergence(normal_b) Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True,
# [0.34939718] with shape: [1] [-1.23893857])
>>> p = normal_a.probs(value_tensor)
>>> print(p)
Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True,
[0.28969154])
>>> kl = normal_a.kl_divergence(normal_b)
>>> print(kl)
Tensor(shape=[1], dtype=float32, place=Place(cpu), stop_gradient=True,
[0.34939718])
""" """
def __init__(self, loc, scale, name=None): def __init__(self, loc, scale, name=None):
......
python/paddle/distribution/transformed_distribution.py
浏览文件 @ 84fe0454
...@@ -29,21 +29,24 @@ class TransformedDistribution(distribution.Distribution): ...@@ -29,21 +29,24 @@ class TransformedDistribution(distribution.Distribution):
.. code-block:: python .. code-block:: python
import paddle >>> import paddle
from paddle.distribution import transformed_distribution >>> paddle.seed(2023)
>>> from paddle.distribution import transformed_distribution
d = transformed_distribution.TransformedDistribution(
paddle.distribution.Normal(0., 1.), >>> d = transformed_distribution.TransformedDistribution(
[paddle.distribution.AffineTransform(paddle.to_tensor(1.), paddle.to_tensor(2.))] ... paddle.distribution.Normal(0., 1.),
) ... [paddle.distribution.AffineTransform(paddle.to_tensor(1.), paddle.to_tensor(2.))]
... )
print(d.sample([10]))
# Tensor(shape=[10], dtype=float32, place=Place(gpu:0), stop_gradient=True, >>> # doctest: +SKIP('random sample')
# [-0.10697651, 3.33609009, -0.86234951, 5.07457638, 0.75925219, >>> print(d.sample([10]))
# -4.17087793, 2.22579336, -0.93845034, 0.66054249, 1.50957513]) Tensor(shape=[10], dtype=float32, place=Place(cpu), stop_gradient=True,
print(d.log_prob(paddle.to_tensor(0.5))) [ 3.22699189, 1.12264419, 0.50283587, 1.83812487, -2.00740123,
# Tensor(shape=[], dtype=float32, place=Place(gpu:0), stop_gradient=True, -2.70338631, 1.26663208, 4.47909021, -0.11529565, 4.32719326])
# -1.64333570) >>> print(d.log_prob(paddle.to_tensor(0.5)))
Tensor(shape=[], dtype=float32, place=Place(cpu), stop_gradient=True,
-1.64333570)
>>> # doctest: -SKIP
""" """
def __init__(self, base, transforms): def __init__(self, base, transforms):
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或