n (int) – An integer indicating the resulting dimension.
size (optional, int, tuple of int) – The batch dimensions of the result.
key (PRNGKey, optional) – The key for the random number generator. If not given, the
default random number generator is used.