mirror of
https://github.com/ml-explore/mlx.git
synced 2025-08-21 12:06:42 +08:00
spelling: bounds
Signed-off-by: Josh Soref <2119212+jsoref@users.noreply.github.com>
This commit is contained in:
parent
c8c2d6c4de
commit
94f71e5832
@ -232,7 +232,7 @@ array truncated_normal(
|
|||||||
auto u = uniform(a, b, shape, dtype, key, s);
|
auto u = uniform(a, b, shape, dtype, key, s);
|
||||||
auto out = multiply(sqrt2, erfinv(u, s), s);
|
auto out = multiply(sqrt2, erfinv(u, s), s);
|
||||||
|
|
||||||
// Clip in bouds
|
// Clip in bounds
|
||||||
return maximum(minimum(upper_t, out, s), lower_t, s);
|
return maximum(minimum(upper_t, out, s), lower_t, s);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user