Add move and swap axis, and vmap for slice, concat, and gather (#158)

* add move and swap axis, and vmap for slice, concat, and gather
2025-09-26 15:58:14 +08:00 · 2023-12-14 12:59:12 -08:00
parent f55908bc48
commit e5851e52b1
10 changed files with 399 additions and 7 deletions
--- a/docs/src/python/ops.rst
+++ b/docs/src/python/ops.rst
@@ -61,6 +61,7 @@ Operations
   mean
   min
   minimum
+   moveaxis
   multiply
   negative
   ones
@@ -87,6 +88,7 @@ Operations
   stop_gradient
   subtract
   sum
+   swapaxes
   take
   take_along_axis
   tan