12
12
</ script >
13
13
14
14
< meta name ="viewport " content ="width=device-width, initial-scale=1.0 " />
15
- < title > dpnp.dpnp_iface_indexing — Data Parallel Extension for NumPy 0.16.0dev0+83.ged95144633 documentation</ title >
15
+ < title > dpnp.dpnp_iface_indexing — Data Parallel Extension for NumPy 0.16.0dev0+109.gb4519755a7 documentation</ title >
16
16
< link rel ="stylesheet " type ="text/css " href ="../../_static/pygments.css?v=fa44fd50 " />
17
17
< link rel ="stylesheet " type ="text/css " href ="../../_static/css/theme.css?v=19f00094 " />
18
18
23
23
24
24
< script src ="../../_static/jquery.js?v=5d32c60e "> </ script >
25
25
< script src ="../../_static/_sphinx_javascript_frameworks_compat.js?v=2cd50e6c "> </ script >
26
- < script src ="../../_static/documentation_options.js?v=c2f7c36d "> </ script >
26
+ < script src ="../../_static/documentation_options.js?v=89825c26 "> </ script >
27
27
< script src ="../../_static/doctools.js?v=9a2dae69 "> </ script >
28
28
< script src ="../../_static/sphinx_highlight.js?v=dc90522c "> </ script >
29
29
< script src ="../../_static/js/theme.js "> </ script >
43
43
Data Parallel Extension for NumPy
44
44
</ a >
45
45
< div class ="version ">
46
- 0.16.0dev0+83.ged95144633
46
+ 0.16.0dev0+109.gb4519755a7
47
47
</ div >
48
48
< div role ="search ">
49
49
< form id ="rtd-search-form " class ="wy-form " action ="../../search.html " method ="get ">
@@ -131,7 +131,7 @@ <h1>Source code for dpnp.dpnp_iface_indexing</h1><div class="highlight"><pre>
131
131
132
132
< span class ="kn "> import</ span > < span class ="nn "> dpctl.tensor</ span > < span class ="k "> as</ span > < span class ="nn "> dpt</ span >
133
133
< span class ="kn "> import</ span > < span class ="nn "> numpy</ span >
134
- < span class ="kn "> from</ span > < span class ="nn "> numpy.core.numeric </ span > < span class ="kn "> import</ span > < span class ="n "> normalize_axis_index</ span >
134
+ < span class ="kn "> from</ span > < span class ="nn "> dpctl.tensor._numpy_helper </ span > < span class ="kn "> import</ span > < span class ="n "> normalize_axis_index</ span >
135
135
136
136
< span class ="kn "> import</ span > < span class ="nn "> dpnp</ span >
137
137
@@ -154,6 +154,7 @@ <h1>Source code for dpnp.dpnp_iface_indexing</h1><div class="highlight"><pre>
154
154
< span class ="s2 "> "diagonal"</ span > < span class ="p "> ,</ span >
155
155
< span class ="s2 "> "extract"</ span > < span class ="p "> ,</ span >
156
156
< span class ="s2 "> "fill_diagonal"</ span > < span class ="p "> ,</ span >
157
+ < span class ="s2 "> "flatnonzero"</ span > < span class ="p "> ,</ span >
157
158
< span class ="s2 "> "indices"</ span > < span class ="p "> ,</ span >
158
159
< span class ="s2 "> "mask_indices"</ span > < span class ="p "> ,</ span >
159
160
< span class ="s2 "> "nonzero"</ span > < span class ="p "> ,</ span >
@@ -613,7 +614,7 @@ <h1>Source code for dpnp.dpnp_iface_indexing</h1><div class="highlight"><pre>
613
614
< span class ="sd "> condition : {array_like, scalar}</ span >
614
615
< span class ="sd "> An array whose non-zero or ``True`` entries indicate the element of `a`</ span >
615
616
< span class ="sd "> to extract.</ span >
616
- < span class ="sd "> a : {dpnp_array , usm_ndarray}</ span >
617
+ < span class ="sd "> a : {dpnp.ndarray , usm_ndarray}</ span >
617
618
< span class ="sd "> Input array of the same size as `condition`.</ span >
618
619
619
620
< span class ="sd "> Returns</ span >
@@ -692,7 +693,7 @@ <h1>Source code for dpnp.dpnp_iface_indexing</h1><div class="highlight"><pre>
692
693
693
694
< span class ="sd "> Parameters</ span >
694
695
< span class ="sd "> ----------</ span >
695
- < span class ="sd "> a : {dpnp_array , usm_ndarray}</ span >
696
+ < span class ="sd "> a : {dpnp.ndarray , usm_ndarray}</ span >
696
697
< span class ="sd "> Array whose diagonal is to be filled in-place. It must be at least 2-D.</ span >
697
698
< span class ="sd "> val : {dpnp.ndarray, usm_ndarray, scalar}</ span >
698
699
< span class ="sd "> Value(s) to write on the diagonal. If `val` is scalar, the value is</ span >
@@ -824,6 +825,55 @@ <h1>Source code for dpnp.dpnp_iface_indexing</h1><div class="highlight"><pre>
824
825
825
826
826
827
828
+ < div class ="viewcode-block " id ="flatnonzero ">
829
+ < a class ="viewcode-back " href ="../../reference/generated/dpnp.flatnonzero.html#dpnp.flatnonzero "> [docs]</ a >
830
+ < span class ="k "> def</ span > < span class ="nf "> flatnonzero</ span > < span class ="p "> (</ span > < span class ="n "> a</ span > < span class ="p "> ):</ span >
831
+ < span class ="w "> </ span > < span class ="sd "> """</ span >
832
+ < span class ="sd "> Return indices that are non-zero in the flattened version of `a`.</ span >
833
+
834
+ < span class ="sd "> This is equivalent to ``dpnp.nonzero(dpnp.ravel(a))[0]``.</ span >
835
+
836
+ < span class ="sd "> For full documentation refer to :obj:`numpy.flatnonzero`.</ span >
837
+
838
+ < span class ="sd "> Parameters</ span >
839
+ < span class ="sd "> ----------</ span >
840
+ < span class ="sd "> a : {dpnp.ndarray, usm_ndarray}</ span >
841
+ < span class ="sd "> Input data.</ span >
842
+
843
+ < span class ="sd "> Returns</ span >
844
+ < span class ="sd "> -------</ span >
845
+ < span class ="sd "> out : dpnp.ndarray</ span >
846
+ < span class ="sd "> Output array, containing the indices of the elements of ``a.ravel()``</ span >
847
+ < span class ="sd "> that are non-zero.</ span >
848
+
849
+ < span class ="sd "> See Also</ span >
850
+ < span class ="sd "> --------</ span >
851
+ < span class ="sd "> :obj:`dpnp.nonzero` : Return the indices of the non-zero elements of</ span >
852
+ < span class ="sd "> the input array.</ span >
853
+ < span class ="sd "> :obj:`dpnp.ravel` : Return a 1-D array containing the elements of</ span >
854
+ < span class ="sd "> the input array.</ span >
855
+
856
+ < span class ="sd "> Examples</ span >
857
+ < span class ="sd "> --------</ span >
858
+ < span class ="sd "> >>> import dpnp as np</ span >
859
+ < span class ="sd "> >>> x = np.arange(-2, 3)</ span >
860
+ < span class ="sd "> >>> x</ span >
861
+ < span class ="sd "> array([-2, -1, 0, 1, 2])</ span >
862
+ < span class ="sd "> >>> np.flatnonzero(x)</ span >
863
+ < span class ="sd "> array([0, 1, 3, 4])</ span >
864
+
865
+ < span class ="sd "> Use the indices of the non-zero elements as an index array to extract</ span >
866
+ < span class ="sd "> these elements:</ span >
867
+
868
+ < span class ="sd "> >>> x.ravel()[np.flatnonzero(x)]</ span >
869
+ < span class ="sd "> array([-2, -1, 1, 2])</ span >
870
+
871
+ < span class ="sd "> """</ span >
872
+
873
+ < span class ="k "> return</ span > < span class ="n "> dpnp</ span > < span class ="o "> .</ span > < span class ="n "> nonzero</ span > < span class ="p "> (</ span > < span class ="n "> dpnp</ span > < span class ="o "> .</ span > < span class ="n "> ravel</ span > < span class ="p "> (</ span > < span class ="n "> a</ span > < span class ="p "> ))[</ span > < span class ="mi "> 0</ span > < span class ="p "> ]</ span > </ div >
874
+
875
+
876
+
827
877
< div class ="viewcode-block " id ="indices ">
828
878
< a class ="viewcode-back " href ="../../reference/generated/dpnp.indices.html#dpnp.indices "> [docs]</ a >
829
879
< span class ="k "> def</ span > < span class ="nf "> indices</ span > < span class ="p "> (</ span >
0 commit comments