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
|
program array_op_2d
implicit none
integer :: a(2, 2, 1), b(2, 2, 1), c(2, 2, 1)
integer :: i, j, k
do i = 1, 2
do j = 1, 2
do k = 1, 1
a(i, j, k) = i*i
b(i, j, k) = j*j
end do
end do
end do
i = 0
print *, a(1, 1, 1), a(1, 2, 1), a(2, 1, 1), a(2, 2, 1)
print *, b(1, 1, 1), b(1, 2, 1), b(2, 1, 1), b(2, 2, 1)
c = a + b
print *, c(1, 1, 1), c(1, 2, 1), c(2, 1, 1), c(2, 2, 1)
if( c(1, 1, 1) /= 2 ) error stop
if( c(1, 2, 1) /= 5 ) error stop
if( c(2, 1, 1) /= 5 ) error stop
if( c(2, 2, 1) /= 8 ) error stop
c = a - b
print *, c(1, 1, 1), c(1, 2, 1), c(2, 1, 1), c(2, 2, 1)
if( c(1, 1, 1) /= 0 ) error stop
if( c(1, 2, 1) /= -3 ) error stop
if( c(2, 1, 1) /= 3 ) error stop
if( c(2, 2, 1) /= 0 ) error stop
c = a*b
print *, c(1, 1, 1), c(1, 2, 1), c(2, 1, 1), c(2, 2, 1)
if( c(1, 1, 1) /= 1 ) error stop
if( c(1, 2, 1) /= 4 ) error stop
if( c(2, 1, 1) /= 4 ) error stop
if( c(2, 2, 1) /= 16 ) error stop
c = b/a
print *, c(1, 1, 1), c(1, 2, 1), c(2, 1, 1), c(2, 2, 1)
if( c(1, 1, 1) /= 1 ) error stop
if( c(1, 2, 1) /= 4 ) error stop
if( c(2, 1, 1) /= 0 ) error stop
if( c(2, 2, 1) /= 1 ) error stop
end program
|
