Question

FORTRAN 矩阵相乘的函数function怎么写？ 必须用 function 不能用 subroutine

貌似function只能返回一个数，如果矩阵相乘要返回二维数组，那么f(i,j)括号里面到底写数组编号还是形参就混乱了，我不知道function怎么写矩阵相乘。

Answer 1

Fortran中本身就具有数组相乘的函数，叫matmul(matrix_a,matrix_b).
--------------------------------------------------
必须要自己写的话可以这样,在主函数中加入函数接口interface，专门针对function返回值为数组的情况。
--------------------------------------------------
program main
implicit none
!!!!定义function的使用接口
interface
function mymatmul(m,n,a,b)
integer:: m, n
real:: a(m,n), b(n,m)
real:: mymatmul(m,m)
end function mymatmul
end interface
real::a(1,2),b(2,1)
data a /1,2/
data b /1,2/
write(*,*)mymatmul(1,2,a,b)
end program main
--------------------------------------------------
function mymatmul(m,n,a,b)
integer:: m, n
real:: a(m,n), b(n,m)
real:: mymatmul(m,m)
integer:: i, j
do i=1,m !行循环
do j=1,m !列循环
mymatmul(i,j)=dot_product( a(i,:), b(:,j) ) !点乘
end do
end do
return
end function mymatmul

Answer 2

可以试试用subroutine，就多加一个call语句嘛。徐士良的书上有，需要的话我可以把代码贴出来。