在动态大小声明后在 fortran 中定义数组

Question

我是 Fortran 新手，我想要的是我的矩阵大小取决于 N 是偶数还是奇数， 天真地我会尝试类似下面的东西，但这不会编译，如果我删除 if 语句它可以正常工作，

 subroutine f(fMatrix,oMatrix,N)
      implicit none; 
      integer,intent(in)::N
      integer::d2
      if(mod(N,2)==0) then ! if N is even then 1 type of size
        parameter(d2=1)
      else
        parameter(d2=2)
      endif
      double precision,intent(in),dimension(N, d2):: fMatrix
      double precision,intent(out),dimension(N, d2):: oMatrix
      !do stuff here with fMatrix
      oMatrix = fMatrix
 end subroutine f

对此有什么可能的解决方法？不去分配？ 我正在使用 f2py，所以这种方式的任何细节都会很方便。

Answer 1

我认为这最接近您想要实现的目标：

subroutine f(fMatrix,oMatrix,N)
  implicit none
  integer,intent(in)                                    :: N
  double precision,intent(in),dimension(N, mod(N,2)+1)  :: fMatrix
  double precision,intent(out),dimension(N, mod(N,2)+1) :: oMatrix

  ! ...
  oMatrix = fMatrix
end subroutine

---

我个人更喜欢以下解决方案之一：

1. 由于fMatrix 和oMatrix 都是虚拟参数 并传递给子例程，您可以使用assumed shape array specifications：

subroutine f(fMatrix,oMatrix,N)
  implicit none
  integer,intent(in)                           :: N
  double precision,intent(in),dimension(:, :)  :: fMatrix
  double precision,intent(out),dimension(:, :) :: oMatrix

  ! ...
end subroutine

现在调用程序需要指定数组的形状。

1. 在外部定义d2并将其传递给子程序：

subroutine f(fMatrix,oMatrix,N, d2)
  implicit none
  integer,intent(in)                            :: N, d2
  double precision,intent(in),dimension(N, d2)  :: fMatrix
  double precision,intent(out),dimension(N, d2) :: oMatrix

  ! ...
end subroutine

并调用子程序：

call f(fMatrix,oMatrix,N,mod(N,2)+1)