Skip to content

Commit 88d1c7d

Browse files
committed
Adding intro to arrayfire example
1 parent 6e2168f commit 88d1c7d

File tree

1 file changed

+66
-0
lines changed

1 file changed

+66
-0
lines changed

examples/getting_started/intro.py

Lines changed: 66 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,66 @@
1+
#!/usr/bin/python
2+
3+
#######################################################
4+
# Copyright (c) 2015, ArrayFire
5+
# All rights reserved.
6+
#
7+
# This file is distributed under 3-clause BSD license.
8+
# The complete license agreement can be obtained at:
9+
# http://arrayfire.com/licenses/BSD-3-Clause
10+
########################################################
11+
12+
import arrayfire as af
13+
import sys
14+
from array import array
15+
16+
if __name__ == "__main__":
17+
if (len(sys.argv) > 1):
18+
af.set_device(int(sys.argv[1]))
19+
af.info()
20+
21+
print("\n---- Intro to ArrayFire using signed(s32) arrays ----\n")
22+
23+
h_A = array('i', ( 1, 2, 4, -1, 2, 0, 4, 2, 3))
24+
h_B = array('i', ( 2, 3, 5, 6, 0, 10,-12, 0, 1))
25+
26+
A = af.Array(h_A, (3,3))
27+
B = af.Array(h_B, (3,3))
28+
29+
print("\n---- Sub referencing and sub assignment\n")
30+
af.display(A)
31+
af.display(A[0,:])
32+
af.display(A[:,0])
33+
A[0,0] = 11
34+
A[1] = 100
35+
af.display(A)
36+
af.display(B)
37+
A[1,:] = B[2,:]
38+
af.display(A)
39+
40+
print("\n---- Bitwise operations\n")
41+
af.display(A & B)
42+
af.display(A | B)
43+
af.display(A ^ B)
44+
45+
print("\n---- Transpose\n")
46+
af.display(A)
47+
af.display(af.transpose(A))
48+
49+
print("\n---- Flip Vertically / Horizontally\n")
50+
af.display(A)
51+
af.display(af.flip(A, 0))
52+
af.display(af.flip(A, 1))
53+
54+
print("\n---- Sum, Min, Max along row / columns\n")
55+
af.display(A)
56+
af.display(af.sum(A, 0))
57+
af.display(af.min(A, 0))
58+
af.display(af.max(A, 0))
59+
af.display(af.sum(A, 1))
60+
af.display(af.min(A, 1))
61+
af.display(af.max(A, 1))
62+
63+
print("\n---- Get minimum with index\n")
64+
(min_val, min_idx) = af.imin(A, 0)
65+
af.display(min_val)
66+
af.display(min_idx)

0 commit comments

Comments
 (0)