Question

fortran function的使用 数组

function function_add(x,y,m,n)
real ::x(m,n),y(m,n),function_add(m,n)

function_add=x+y
end

、、、、、、、、、、、、、、、、、、、、、、、
program sub

implicit none
real :: a(2,2),b(2,2)
a=reshape((/1,1,2,3/),(/2,2/))
b=reshape((/2,2,2,2/),(/2,2/))
print *,function_add(a,b,2,2)
end program sub

就是想fuction返回数组怎么写呀

Answer 1

一般情况下，不建议把数组作为function的返回值。一般的做法是用subroutine，把数组作为参数传递引用。
如果一定要使用function传数组返回值，可以把函数嵌到调用函数中作为一个子函数。
program sub

implicit none
real :: a(2,2),b(2,2), arr(2,2)
a=reshape((/1,1,2,3/),(/2,2/))
b=reshape((/2,2,2,2/),(/2,2/))
print*, function_add(a,b,2,2)

contains
function function_add(x,y,m,n)
implicit none
integer:: m, n
real :: x(m,n), y(m,n), function_add(m,n)

function_add = x + y
end function function_add

end program sub

Answer 2

下面是我写的一个矩阵相乘的函数(比库函数matmul()少了一点功能)，你可以参考一下(要放在模块里，不然可能会报错，放主程序里要用接口)
function matrix_multiply (A, B)
implicit none
real :: A(:, :), B(:, :)
real, allocatable :: matrix_multiply(:, :)
integer n1, n2, n3, n4
integer i, j, k
n1 = size (A, 1)
n2 = size (A, 2)
n3 = size (B, 1)
n4 = size (B, 2)

if (n2 /= n3) then
write (*, *) "the demension is not match!!"
stop
else
allocate (matrix_multiply(n1, n4))
end if

do i = 1, n1
do j = 1, n4
matrix_multiply (i, j) = 0.0
do k = 1, n2
matrix_multiply(i, j) = matrix_multiply(i, j) + A(i, k) * B(k, j)
end do
end do
end do

return
end function matrix_multiply

其实矩阵相加没必要写个函数，a+b就好了