File tree Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Expand file tree Collapse file tree 2 files changed +2
-3
lines changed Original file line number Diff line number Diff line change @@ -149,7 +149,6 @@ def one_hot(
149
149
An array having the same shape as `x` except for a new axis at the position
150
150
given by `axis` having size `num_classes`.
151
151
152
-
153
152
If ``x < 0`` or ``x >= num_classes``, then the result is undefined, may raise
154
153
an exception, or may even cause a bad state. `x` is not checked.
155
154
Original file line number Diff line number Diff line change @@ -388,8 +388,8 @@ def one_hot(
388
388
x_size : int ,
389
389
dtype : DType ,
390
390
xp : ModuleType ,
391
- ) -> Array :
392
- """Helper for _delegation.one_hot ."""
391
+ ) -> Array : # numpydoc ignore=PR01,RT01
392
+ """See docstring in `array_api_extra. _delegation.py` ."""
393
393
out = xp .zeros ((x .size , num_classes ), dtype = dtype )
394
394
x_flattened = xp .reshape (x , (- 1 ,))
395
395
if is_numpy_namespace (xp ):
You can’t perform that action at this time.
0 commit comments