mlx_random_nif (mlx v0.2.0)

View Source

Summary

Functions

bernoulli(P, Shape, Dtype)

-spec bernoulli(reference(), [integer()], atom()) -> {ok, reference()} | {error, term()}.

beta(Alpha, Beta, Shape, Dtype)

-spec beta(reference(), reference(), [integer()], atom()) -> {ok, reference()} | {error, term()}.

categorical(Logits, Shape, Dtype)

-spec categorical(reference(), [integer()], atom()) -> {ok, reference()} | {error, term()}.

choice(Array, Size, Replace)

-spec choice(reference(), integer(), boolean()) -> {ok, reference()} | {error, term()}.

exponential(Lambda, Shape, Dtype)

-spec exponential(reference(), [integer()], atom()) -> {ok, reference()} | {error, term()}.

gamma(Alpha, Beta, Shape, Dtype)

-spec gamma(reference(), reference(), [integer()], atom()) -> {ok, reference()} | {error, term()}.

key(Seed)

-spec key(integer()) -> {ok, reference()} | {error, term()}.

multinomial(P, N, Shape, Dtype)

-spec multinomial(reference(), integer(), [integer()], atom()) -> {ok, reference()} | {error, term()}.

normal(Shape, Mean, Std)

-spec normal([integer()], number(), number()) -> {ok, reference()} | {error, term()}.

permutation(N)

-spec permutation(integer()) -> {ok, reference()} | {error, term()}.

permutation(Array, Axis)

-spec permutation(reference(), integer()) -> {ok, reference()} | {error, term()}.

poisson(Lambda, Shape, Dtype)

-spec poisson(reference(), [integer()], atom()) -> {ok, reference()} | {error, term()}.

randint(Low, High, Shape, Dtype)

-spec randint(integer(), integer(), [integer()], atom()) -> {ok, reference()} | {error, term()}.

seed(Seed)

-spec seed(integer()) -> ok | {error, term()}.

shuffle(Array, Axis)

-spec shuffle(reference(), integer()) -> {ok, reference()} | {error, term()}.

uniform(Shape, Low, High, Dtype)

-spec uniform([integer()], number(), number(), atom()) -> {ok, reference()} | {error, term()}.