brainpy.math.random.RandomState
brainpy.math.random.RandomState#
- class brainpy.math.random.RandomState(seed_or_key=None, seed=None)[source]#
RandomState that track the random generator state.
- __init__(seed_or_key=None, seed=None)[source]#
RandomState constructor.
- Parameters
seed_or_key (int, Array, optional) –
It can be an integer for initial seed of the random number generator, or it can be a JAX’s PRNKey, which is an array with two elements and uint32 dtype.
New in version 2.2.3.4.
seed (int, ArrayType, optional) –
Same as seed_or_key.
Deprecated since version 2.2.3.4: Will be removed since version 2.4.
Methods
__init__
([seed_or_key, seed])RandomState constructor.
all
([axis, keepdims])Returns True if all elements evaluate to True.
any
([axis, keepdims])Returns True if any of the elements of a evaluate to True.
argmax
([axis])Return indices of the maximum values along the given axis.
argmin
([axis])Return indices of the minimum values along the given axis.
argpartition
(kth[, axis, kind, order])Returns the indices that would partition this array.
argsort
([axis, kind, order])Returns the indices that would sort this array.
as_variable
()As an instance of Variable.
astype
(dtype)Copy of the array, cast to a specified type.
bernoulli
(p[, size, key])beta
(a, b[, size, key])binomial
(n, p[, size, key])block_host_until_ready
(*args)block_until_ready
(*args)byteswap
([inplace])Swap the bytes of the array elements
categorical
(logits[, axis, size, key])chisquare
(df[, size, key])choice
(a[, size, replace, p, key])choose
(choices[, mode])Use an index array to construct a new array from a set of choices.
clip
([min, max])Return an array whose values are limited to [min, max].
clone
()compress
(condition[, axis])Return selected slices of this array along given axis.
conj
()Complex-conjugate all elements.
conjugate
()Return the complex conjugate, element-wise.
copy
()Return a copy of the array.
cumprod
([axis, dtype])Return the cumulative product of the elements along the given axis.
cumsum
([axis, dtype])Return the cumulative sum of the elements along the given axis.
device
()diagonal
([offset, axis1, axis2])Return specified diagonals.
dirichlet
(alpha[, size, key])dot
(b)Dot product of two arrays.
exponential
([scale, size, key])f
(dfnum, dfden[, size, key])fill
(value)Fill the array with a scalar value.
flatten
([order])gamma
(shape[, scale, size, key])geometric
(p[, size, key])gumbel
([loc, scale, size, key])hypergeometric
(ngood, nbad, nsample[, size, key])item
(*args)Copy an element of an array to a standard Python scalar and return it.
laplace
([loc, scale, size, key])loggamma
(a[, size, key])logistic
([loc, scale, size, key])lognormal
([mean, sigma, size, key])logseries
(p[, size, key])max
([axis, keepdims])Return the maximum along a given axis.
maxwell
([size, key])mean
([axis, dtype, keepdims])Returns the average of the array elements along given axis.
min
([axis, keepdims])Return the minimum along a given axis.
multinomial
(n, pvals[, size, key])multivariate_normal
(mean, cov[, size, ...])negative_binomial
(n, p[, size, key])noncentral_chisquare
(df, nonc[, size, key])noncentral_f
(dfnum, dfden, nonc[, size, key])nonzero
()Return the indices of the elements that are non-zero.
normal
([loc, scale, size, key])numpy
([dtype])Convert to numpy.ndarray.
orthogonal
(n[, size, key])pareto
(a[, size, key])permutation
(x[, axis, independent, key])poisson
([lam, size, key])power
(a[, size, key])prod
([axis, dtype, keepdims, initial, where])Return the product of the array elements over the given axis.
ptp
([axis, keepdims])Peak to peak (maximum - minimum) value along a given axis.
put
(indices, values)Replaces specified elements of an array with given values.
rand
(*dn[, key])randint
(low[, high, size, dtype, key])randn
(*dn[, key])random
([size, key])random_integers
(low[, high, size, key])random_sample
([size, key])ranf
([size, key])ravel
([order])Return a flattened array.
rayleigh
([scale, size, key])repeat
(repeats[, axis])Repeat elements of an array.
reshape
(*shape[, order])Returns an array containing the same data with a new shape.
resize
(new_shape)Change shape and size of array in-place.
round
([decimals])Return
a
with each element rounded to the given number of decimals.sample
([size, key])searchsorted
(v[, side, sorter])Find indices where elements should be inserted to maintain order.
seed
([seed_or_key, seed])Sets a new random seed.
shuffle
(x[, axis, key])sort
([axis, kind, order])Sort an array in-place.
split
(indices_or_sections[, axis])Split an array into multiple sub-arrays as views into
ary
.split_key
()Create a new seed from the current seed.
split_keys
(n)Create multiple seeds from the current seed.
squeeze
([axis])Remove axes of length one from
a
.standard_cauchy
([size, key])standard_exponential
([size, key])standard_gamma
(shape[, size, key])standard_normal
([size, key])standard_t
(df[, size, key])std
([axis, dtype, ddof, keepdims])Compute the standard deviation along the specified axis.
sum
([axis, dtype, keepdims, initial, where])Return the sum of the array elements over the given axis.
swapaxes
(axis1, axis2)Return a view of the array with axis1 and axis2 interchanged.
t
(df[, size, key])take
(indices[, axis, mode])Return an array formed from the elements of a at the given indices.
tile
(reps)Construct an array by repeating A the number of times given by reps.
to_jax
([dtype])Convert to jax.numpy.ndarray.
to_numpy
([dtype])Convert to numpy.ndarray.
tobytes
([order])Construct Python bytes containing the raw data bytes in the array.
tolist
()Return the array as an
a.ndim
-levels deep nested list of Python scalars.trace
([offset, axis1, axis2, dtype])Return the sum along diagonals of the array.
transpose
(*axes)Returns a view of the array with axes transposed.
triangular
([size, key])truncated_normal
(lower, upper, size[, ...])uniform
([low, high, size, key])update
(value)Update the value of this Array.
var
([axis, dtype, ddof, keepdims])Returns the variance of the array elements, along given axis.
view
([dtype])New view of array with the same data.
vonmises
(mu, kappa[, size, key])wald
(mean, scale[, size, key])weibull
(a[, size, key])weibull_min
(a[, scale, size, key])Sample from a Weibull minimum distribution.
zipf
(a[, size, key])Attributes
T
at
batch_axis
batch_size
device_buffer
dtype
Variable dtype.
imag
is_brainpy_array
ndim
real
shape
Variable shape.
shape_nb
Shape without batch axis.
size
value