GV

GeezerVison

The World Wide Web, restored to 800 pixels of glory.

Home|About|Gaming cabinet|Archives|How it works|Starter sites

On this site

  • Sign in
  • ActionsCI/CD and automation
  • CodespacesCloud development environments
  • All Features
  • Documentation
  • Training
  • Certifications
  • Build with Copilot
  • Customize Copilot
  • Integrations
  • Trending repos
  • GitHub Sponsors
  • Developer articles
  • Enterprises

This view

  • Original page
  • New address

GeezerVison is showing github.com — arrayfire-py/arrayfire/array_api/_searching_functions.py at master · arrayfire/arrayfire-py

Expand file tree

/

_searching_functions.py

Copy path
More file actions
More file actions

Latest commit

 

History

History
History
71 lines (52 loc) · 2.27 KB
/

_searching_functions.py

Copy path

File metadata and controls

71 lines (52 loc) · 2.27 KB
Raw
Copy raw file
Download raw file
Open symbols panel
Edit and raw actions
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
from __future__ import annotations
import arrayfire as af
from ._array_object import Array
def argmax(x: Array, /, *, axis: int | None = None, keepdims: bool = False) -> Array:
# TODO
# [] Add documentation
# [] Figure out what to do with keepdims arg that is not actually works well with af case
# [] Fix typings
# source: https://data-apis.org/array-api/2022.12/API_specification/generated/array_api.argmax.html#argmax
if axis is None:
flat_array = af.flat(x._array)
_, indices = af.imax(flat_array, axis=0)
else:
_, indices = af.imax(x._array, axis=axis)
if keepdims:
shape = tuple([1] * x.ndim)
indices = af.moddims(indices, shape) # type: ignore[arg-type] # FIXME
return Array._new(indices)
def argmin(x: Array, /, *, axis: int | None = None, keepdims: bool = False) -> Array:
# TODO
# [] Add documentation
# [] Figure out what to do with keepdims arg that is not actually works well with af case
# [] Fix typings
# source: https://data-apis.org/array-api/2022.12/API_specification/generated/array_api.argmin.html#argmin
if axis is None:
flat_array = af.flat(x._array)
_, indices = af.imin(flat_array, axis=0)
else:
_, indices = af.imin(x._array, axis=axis)
if keepdims:
shape = tuple([1] * x.ndim)
indices = af.moddims(indices, shape) # type: ignore[arg-type] # FIXME
return Array._new(indices)
def nonzero(x: Array, /) -> tuple[Array, ...]:
# TODO
# Add documentation
# source: https://data-apis.org/array-api/2022.12/API_specification/generated/array_api.nonzero.html#nonzero
flat_array = af.flat(x._array)
non_zero_indices = af.where(flat_array != 0)
if len(x.shape) == 1:
return (Array._new(non_zero_indices),)
else:
idx = []
for dim in reversed(x.shape):
idx.append(Array._new(non_zero_indices % dim))
non_zero_indices = non_zero_indices // dim
return tuple(reversed(idx))
def where(condition: Array, x1: Array, x2: Array, /) -> Array:
# TODO
# Add documentation
# source: https://data-apis.org/array-api/2022.12/API_specification/generated/array_api.where.html#where
return Array._new(af.select(x1._array, x2._array, condition._array))
You are visitor 01402520 — GeezerVison does not store the pages you look up.
Best viewed with Netscape Navigator 4.7 or Internet Explorer 6 at 800×600. No cookies required.