Skip to content

Commit 0e6ec5d

Browse files
committed
corr_dev: clarification
1 parent 1fdabd9 commit 0e6ec5d

File tree

1 file changed

+12
-16
lines changed

1 file changed

+12
-16
lines changed

src/stdlib_experimental_stats_corr.fypp

Lines changed: 12 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -200,8 +200,8 @@ contains
200200

201201
select case(dim)
202202
case(1)
203-
do i = 1, size(x, 2)
204-
do j = 1, size(x, 2)
203+
do i = 1, size(res, 2)
204+
do j = 1, size(res, 1)
205205
mask_ = merge(.true., .false., mask(:, i) .and. mask(:, j))
206206
centeri_ = merge( x(:, i) - mean(x(:, i), mask = mask_),&
207207
#:if t1[0] == 'r'
@@ -225,8 +225,8 @@ contains
225225
end do
226226
end do
227227
case(2)
228-
do i = 1, size(x, 1)
229-
do j = 1, size(x, 1)
228+
do i = 1, size(res, 2)
229+
do j = 1, size(res, 1)
230230
mask_ = merge(.true., .false., mask(i, :) .and. mask(j, :))
231231
centeri_ = merge( x(i, :) - mean(x(i, :), mask = mask_),&
232232
#:if t1[0] == 'r'
@@ -272,15 +272,13 @@ contains
272272

273273
select case(dim)
274274
case(1)
275-
do i = 1, size(x, 2)
276-
do j = 1, size(x, 2)
275+
do i = 1, size(res, 2)
276+
do j = 1, size(res, 1)
277277
mask_ = merge(.true., .false., mask(:, i) .and. mask(:, j))
278278
centeri_ = merge( x(:, i) - mean(x(:, i), mask = mask_),&
279-
0._dp,&
280-
mask_)
279+
0._dp, mask_)
281280
centerj_ = merge( x(:, j) - mean(x(:, j), mask = mask_),&
282-
0._dp,&
283-
mask_)
281+
0._dp, mask_)
284282

285283
res(j, i) = dot_product( centerj_, centeri_)&
286284
/sqrt(dot_product( centeri_, centeri_)*&
@@ -289,15 +287,13 @@ contains
289287
end do
290288
end do
291289
case(2)
292-
do i = 1, size(x, 1)
293-
do j = 1, size(x, 1)
290+
do i = 1, size(res, 2)
291+
do j = 1, size(res, 1)
294292
mask_ = merge(.true., .false., mask(i, :) .and. mask(j, :))
295293
centeri_ = merge( x(i, :) - mean(x(i, :), mask = mask_),&
296-
0._dp,&
297-
mask_)
294+
0._dp, mask_)
298295
centerj_ = merge( x(j, :) - mean(x(j, :), mask = mask_),&
299-
0._dp,&
300-
mask_)
296+
0._dp, mask_)
301297

302298
res(j, i) = dot_product( centerj_, centeri_)&
303299
/sqrt(dot_product( centeri_, centeri_)*&

0 commit comments

Comments
 (0)