FilesExpand file tree

 main

/

array_expr_01.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

37 lines (30 loc) · 1002 Bytes

 main

/

array_expr_01.py

Copy path

Top

File metadata and controls

• Code
• Blame

37 lines (30 loc) · 1002 Bytes

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

from lpython import Const, i32, f32, f64

from numpy import empty, reshape, int32, float64

def array_expr_01():

dim1: Const[i32] = 10

dim2: Const[i32] = 10

dim3: Const[i32] = 5

dim1d: Const[i32] = dim1 * dim2 * dim3

i: i32

shape1d: i32[1] = empty(1, dtype=int32)

shape3d: i32[3] = empty(3, dtype=int32)

eps: f64

eps = 1e-12

e: f64[10, 10, 5] = empty((dim1, dim2, dim3), dtype=float64)

f: f64[10, 10, 5] = empty((dim1, dim2, dim3), dtype=float64)

g: f64[500] = empty(dim1d, dtype=float64)

e1d: f64[500] = empty((dim1d), dtype=float64)

f1d: f64[500] = empty((dim1d), dtype=float64)

for i in range(dim1d):

e1d[i] = float(i + 1)

f1d[i] = float(i + 1)

shape3d[0] = dim1

shape3d[1] = dim2

shape3d[2] = dim3

shape1d[0] = dim1d

e = reshape(e1d, shape3d)

f = reshape(f1d, shape3d)

g = reshape(e + f, shape1d)

for i in range(dim1d):

assert abs(g[i] - f64(2*(i + 1))) <= eps

array_expr_01()