Put along axis + fixe for partition grad (#1430)

* put along axis, fixes for partition grad * zeros for arg reduce
2025-09-03 14:24:44 +08:00 · 2024-09-23 10:03:38 -07:00
parent 2b878e9dd7
commit 195b429d99
9 changed files with 220 additions and 9 deletions
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -121,6 +121,7 @@ Operations
   pad
   power
   prod
+   put_along_axis
   quantize
   quantized_matmul
   radians