Discussion :

[domdom974]

Bonjour,

j'essaye actuellement de faire un produit de Kronecker en fortran 90.

Voici le code de ma fonction:

MODULE kronec

!
! Calcul du produit de kronecker par la fonction kron
!

!IMPLICIT NONE


CONTAINS

FUNCTION kron(A,B)

IMPLICIT NONE

!!
!!! Calcul du produit de kronecker, equivalent de la fonction kron matlab
!!


REAL, dimension( : , : ) , intent(in) :: A,B


REAL, dimension(size(B,1),size(B,2),size(A,1),size(A,2)) :: C
REAL, dimension(size(A,1)*size(A,2),size(B,1)*size(B,2)) :: kron

INTEGER :: i, j, m, n, p, q, ii, jj
INTEGER :: ind1, ind2, ip, jq, mp, nq


m = size(A,1)
n = size(A,2)
p = size(B,1)
q = size(B,2)

!m=2
!n=3
!p=4
!q=5


mp = m*p
nq = n*q


DO i = 1,m
DO j = 1,n
DO ii = 1,p
DO jj = 1,q
C(ii,jj,i,j) = A(i,j)*B(ii,jj)
END DO
END DO
END DO
END DO


DO i = 1,m
ind1 = (i-1)*p+1
ip = i*p
DO j = 1,n
ind2 = (j-1)*q+1
jq = j*q
kron(ind1:ip,ind2:jq) = C(:,:,i,j)
END DO
END DO



WRITE(*, *) 'C=', C

WRITE(*, *) 'C(:,:,1,1)=',C(:,:,1,1)



END FUNCTION kron



END MODULE kronec

Je sais que je calcule bien ce qu'il faut avec le tableau C. Cependant lorsque je veux ranger les valeurs dans la matrice kron, il y a un problème: à partir de la deuxième colonne (jusqu'à la fin) et pour les deux premières lignes le récupère les valeurs sur les deux dernières lignes depuis la première colonne jusqu'à l'avant dernière.

Donc je sais que c'est un problème d'indice, mais je n'arrive pas à le résoudre.

Si quelqu'un à une idée, je suis preneur, merci d'avance

[doguette]

Bonjour,
Voici une subroutine que j'avais écrite il y a quelque temps, sans garantie:

Code :

Sélectionner tout - Visualiser dans une fenêtre à part

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
subroutine kron_prod(tab,tab2,tab3)
  ! Kronecker product
  Implicit none
  Integer :: i,j,k,l,m,n,o,p
  Real, dimension(:,:) :: tab, tab2
  real, allocatable:: tab3(:,:)
  ! Matrix tab of size (m*n), matrix tab2 of size (o*p)
  m=size(tab, dim=1)
  n=size(tab, dim=2)
  o=size(tab2, dim=1)
  p=size(tab2, dim=2)
  Allocate(tab3(m*o , n*p))
  tab3=0
  Do i=1,m  ! tab(i,.)
        Do j=1,n  ! tab(i,j)
              Do k=1,o
                    Do l=1,p
                           tab3((i-1)*o+k,(j-1)*p+l)=tab(i,j)*tab2(k,l)
                     Enddo
               Enddo
         Enddo
  enddo
end subroutine kron_prod

En espérant que cela puisse t'aider...