Fortran90数组段运算：如何实现一维数组与多维数组维度匹配？

嘿，这个问题我之前也碰到过！Fortran的数组运算对形状匹配要求很严格，直接用一维数组b和多维数组a相乘会报错，但完全不用复制b成大数组浪费内存，这里有几个高效的解决方案：

1. 利用Fortran 2008的自动广播（推荐） #

从Fortran 2008开始，主流编译器（比如gfortran、Intel Fortran）都支持隐式形状广播——只要让b的维度和a的对应维度对齐，编译器会自动“逻辑扩展”b的维度，不会实际创建临时数组占用额外内存。

你只需要把b转换成一个三维数组，让它的第一和第三维度大小为1（和a的对应维度匹配），就可以直接相乘：

real*8 a(20,5,2)
real*8 b(5)
real*8 c(20,5,2)
! 初始化a、b的值
a = 1.0d0
b = [1.0d0, 2.0d0, 3.0d0, 4.0d0, 5.0d0]
! 用reshape调整b的形状，触发广播
c = a * reshape(b, [1, 5, 1])

这里reshape不会复制b的内存（编译器会优化掉冗余复制），广播过程也只是逻辑上的维度扩展，内存占用还是b原本的5个元素大小。

如果你觉得reshape不够直观，也可以用spread函数显式指定扩展维度，效果是一样的：

! 先在第一维度扩展b为(20,5)，再在第三维度扩展为(20,5,2)
c = a * spread(spread(b, dim=1, ncopies=20), dim=3, ncopies=2)

2. 用隐含do循环（兼容旧标准） #

如果你需要兼容Fortran 90/95这类旧标准，不想依赖2008的特性，直接写嵌套循环就行。别担心效率，现代编译器会自动把循环向量化优化，性能和数组运算几乎没差别，而且完全不需要额外内存：

real*8 a(20,5,2)
real*8 b(5)
real*8 c(20,5,2)
! 初始化a、b的值
do k = 1, 2
  do i = 1, 20
    c(i,:,k) = a(i,:,k) * b
  end do
end do

3. 使用forall语句（并行化赋值） #

另一种兼容旧标准的方法是forall语句，它本质上是并行化的数组赋值语法，写法更简洁，同样不会产生临时数组：

real*8 a(20,5,2)
real*8 b(5)
real*8 c(20,5,2)
! 初始化a、b的值
forall(i=1:20, k=1:2)
  c(i,:,k) = a(i,:,k) * b
end forall

为什么这些方法不浪费内存？ #

• 广播方案：编译器只会在运算时逻辑扩展b的维度，不会实际创建和a一样大的临时数组，内存占用始终是b的5个元素。
• 循环/forall方案：直接逐个元素计算，完全不需要额外内存存储扩展后的b。

内容的提问来源于stack exchange，提问作者bob.sacamento