We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 620372c commit f49f907Copy full SHA for f49f907
src/array_api_extra/_delegation.py
@@ -137,8 +137,8 @@ def one_hot(
137
dtype : DType, optional
138
The dtype of the return value. Defaults to the default float dtype (usually
139
float64).
140
- axis : int or tuple of ints, optional
141
- Position(s) in the expanded axes where the new axis is placed.
+ axis : int, optional
+ Position in the expanded axes where the new axis is placed. Default: -1.
142
xp : array_namespace, optional
143
The standard-compatible namespace for `x`. Default: infer.
144
0 commit comments