docs: add take_along_axis signatures

kgryte · kgryte · commit f61360fccc8b · 2024-05-30T01:45:28.000-07:00
diff --git a/signatures/indexing/take_along_axis.md b/signatures/indexing/take_along_axis.md
@@ -0,0 +1,53 @@
+# take_along_axis
+
+## NumPy
+
+```
+numpy.take_along_axis(arr, indices, axis) → ndarray
+```
+
+When `axis` is `None`, flatten for consistency with `sort` and `argsort`.
+
+## CuPy
+
+```
+cupy.take_along_axis(a, indices, axis) → ndarray
+```
+
+## dask.array
+
+```
+
+```
+
+- <https://github.com/dask/dask/issues/3663>
+- <https://github.com/dask/dask/pull/11076>
+
+## JAX
+
+```
+jax.numpy.take_along_axis(arr, indices, axis, mode=None, fill_value=None) → ndarray
+```
+
+- "mode" specifies how to handle out-of-bounds indices.
+- "fill_value" specifies the value to use when "mode" is "fill"
+
+## MXNet
+
+```
+
+```
+
+## PyTorch
+
+```
+torch.take_along_dim(input, indices, dim=None, *, out=None) → Tensor
+```
+
+Supports flattening when `dim=None`.
+
+## TensorFlow
+
+```
+tf.experimental.numpy.take_along_axis(arr, indices, axis) → Tensor
+```
