Add missing funcs to docs (#2021)

This commit is contained in:
Stephen Panaro
2025-03-30 21:29:33 -04:00
committed by GitHub
parent 5f5770e3a2
commit 90823d2938
3 changed files with 4 additions and 1 deletions

View File

@@ -467,7 +467,7 @@ void init_random(nb::module_& parent_module) {
array: The output array of random values.
)pbdoc");
m.def(
"permuation",
"permutation",
[](const std::variant<nb::int_, mx::array>& x,
int axis,
const std::optional<mx::array>& key_,