请问fortran有办法实现数组维数是个变量吗？

statdog · 发表于 2020-5-24 02:30:38

目前在做一个tensor data的项目，要写一个R package, 导师建议我用fortran写核心运算部分。目前在造轮子，运算需要处理不同维数的数组，所以我希望一个子程序可以同时处理3，4，5维数组。但是我查的资料都说，在声明数组的时候必须指定数组维数。想求助一下各位大大，在fortran中有没有方法可以让子程序的数组参数的维数不是一个constant？十分感谢！

vvt · 发表于 2020-5-24 12:51:30

1. 如果你的多维数组不同元素直接是独立的，毫不相干。那么可以尝试用“逐元函数”，比如语法内的 sin 函数，就是一个逐元函数。你可以直接用它处理一个多维数组，计算每一个元素的正弦值。
2. 如果你的多维数组之间不是独立的，相互联系。你只能写多份函数，通过generic name让他们重载成一个通用名的函数名。比如语法内的 sin 函数，它同时也是重载的。
你可以用它处理单精度数据，也可以用它处理双精度数据。其实编译器的运行时库就是写了2份代码（sin和dsin）。

statdog · 发表于 2020-5-24 14:15:42

vvt 发表于 2020-5-24 12:51
1. 如果你的多维数组不同元素直接是独立的，毫不相干。那么可以尝试用“逐元函数”，比如语法内的 sin 函数 ...

感谢。我暂时用了select case对3到5每个case写了一遍函数。

statdog · 发表于 2020-5-24 14:20:54

vvt 发表于 2020-5-24 12:51
1. 如果你的多维数组不同元素直接是独立的，毫不相干。那么可以尝试用“逐元函数”，比如语法内的 sin 函数 ...

目前还遇到一个问题，就是我想对子程序的几个argument写一个循环。
比如subroutine(x,y, arg1,arg2,arg3,arg4,arg5)，我想有一个循环从i=1到5每次对argi做相同的处理。请问大大这样应该怎么实现呢？感谢！

li913 · 发表于 2020-5-24 15:05:35

statdog 发表于 2020-5-24 14:20
目前还遇到一个问题，就是我想对子程序的几个argument写一个循环。
比如subroutine(x,y, arg1,arg2,arg3, ...

如果参数类型一致，用数组；否则，手动处理。

vvt · 发表于 2020-5-24 16:42:28

statdog 发表于 2020-5-24 14:15
感谢。我暂时用了select case对3到5每个case写了一遍函数。

那你岂不是得写3个参数？？

statdog · 发表于 2020-5-24 21:17:35

li913 发表于 2020-5-24 15:05
如果参数类型一致，用数组；否则，手动处理。

这些参数是不同维数的矩阵，用数组的话，把它们拉直放在一个数组里变成一个参数确实可以处理。我试一下。感谢！

statdog · 发表于 2020-5-24 21:21:54

vvt 发表于 2020-5-24 16:42
那你岂不是得写3个参数？？

我把输入的tensor拉直了，在不同的case里分别reshape回去再拉直，这样似乎没有问题。

[子程序] 请问fortran有办法实现数组维数是个变量吗？

规矩勋章

元老勋章

新人勋章

水王勋章

热心勋章