Add LogNormalRV

aesara/tensor/random/basic.py

@@ -74,6 +74,20 @@ class HalfNormalRV(RandomVariable):
 halfnormal = HalfNormalRV()
 
 
+class LogNormalRV(RandomVariable):
+    name = "lognormal"
+    ndim_supp = 0
+    ndims_params = [0, 0]
+    dtype = "floatX"
+    _print_name = ("LogN", "\\operatorname{LogN}")
+
+    def __call__(self, mean=0.0, sigma=1.0, size=None, **kwargs):
+        return super().__call__(mean, sigma, size=size, **kwargs)
+
+
+lognormal = LogNormalRV()
+
+
 class GammaRV(RandomVariable):
     name = "gamma"
     ndim_supp = 0

tests/tensor/random/test_basic.py

@@ -30,6 +30,7 @@ from aesara.tensor.random.basic import (
     hypergeometric,
     invgamma,
     laplace,
+    lognormal,
     multinomial,
     multivariate_normal,
     nbinom,
@@ -213,6 +214,15 @@ def test_halfnormal_samples():
     )
 
 
+def test_lognormal_samples():
+    test_mean = np.array(0, dtype=config.floatX)
+    test_sigma = np.array(1, dtype=config.floatX)
+
+    rv_numpy_tester(lognormal)
+    rv_numpy_tester(lognormal, test_mean)
+    rv_numpy_tester(lognormal, test_mean, test_sigma, size=[2, 3])
+
+
 def test_gamma_samples():
     test_a = np.array(0.5, dtype=config.floatX)
     test_b = np.array(0.5, dtype=config.floatX)