def testPareto(self, b, dtype): key = random.PRNGKey(0) rand = lambda key, b: random.pareto(key, b, (10000, ), dtype) crand = api.jit(rand) uncompiled_samples = rand(key, b) compiled_samples = crand(key, b) for samples in [uncompiled_samples, compiled_samples]: self._CheckKolmogorovSmirnovCDF(samples, scipy.stats.pareto(b).cdf)
def testParetoShape(self): key = random.PRNGKey(0) x = random.pareto(key, onp.array([0.2, 0.3]), shape=(3, 2)) assert x.shape == (3, 2)
def pareto(a, size=None): return JaxArray( jr.pareto(DEFAULT.split_key(), b=a, shape=_size2shape(size)))
def _rvs(self, b): return random.pareto(self._random_state, b, shape=self._size)
def pareto(self, a, size=None): return JaxArray( jr.pareto(self.split_key(), b=a, shape=_size2shape(size)))