Function
argmax
(a, axis=None, keepdims=False, split_every=None, out=None)
Source from the content-addressed store, hash-verified
| 931 | |
| 932 | @derived_from(np) |
| 933 | def argmax(a, axis=None, keepdims=False, split_every=None, out=None): |
| 934 | return arg_reduction( |
| 935 | a, |
| 936 | partial(arg_chunk, chunk.max, chunk.argmax), |
| 937 | partial(arg_combine, chunk.argmax), |
| 938 | partial(arg_agg, chunk.argmax), |
| 939 | axis=axis, |
| 940 | keepdims=keepdims, |
| 941 | split_every=split_every, |
| 942 | out=out, |
| 943 | ) |
| 944 | |
| 945 | |
| 946 | @derived_from(np) |
Tested by
no test coverage detected