我想用SymPy codegen实用程序生成一个Fortran子例程。我可以生成一个Fortran函数而不会出现问题codegen(("f", x*y*z), "f95", "filename")。但是我想生成一个Fortran子例程，以便可以修改输入数组。我怎样才能做到这一点？该文档非常差。
如果只有一个标量返回值，则codegen实用程序将创建一个函数，否则将创建一个子例程。有一些对数组的支持，但是除非您向codegen提供类似表达式的数组，否则不会触发数组功能。文档是分散的，因此我将给您一些提示：
看一下自动包装文档中的matrix-vector示例：http : //docs.sympy.org/latest/modules/utilities/autowrap.html。Autowrap在后台使用了代码生成器。
在当前的开发人员版本中，还具有用于生成与具有符号元素的矩阵相对应的代码的功能。请参阅http://docs.sympy.org/dev/modules/printing.html#fortran-printing上fcode（）的示例。
以下示例代码应为矩阵向量乘积输出一个Fortran 95子例程：
from sympy import *
from sympy.utilities.codegen import codegen
A, B, C = symbols('A B C', cls=IndexedBase)
m, n = symbols('m n', integer=True)
i = Idx('i', m)
j = Idx('j', n)
expr = Eq(C[i], A[i, j]*B[j])
result = codegen(('my_function', expr), 'f95', 'my_project')
print result[0][1]
通过将这些行保存到my_file.py并运行python my_file.py，我得到以下输出：
!******************************************************************************
!* Code generated with sympy 0.7.5-git *
!* *
!* See http://www.sympy.org/ for more information. *
!* *
!* This file is part of 'project' *
!******************************************************************************
subroutine my_function(A, B, m, n, C)
implicit none
INTEGER*4, intent(in) :: m
INTEGER*4, intent(in) :: n
REAL*8, intent(in), dimension(1:m, 1:n) :: A
REAL*8, intent(in), dimension(1:n) :: B
REAL*8, intent(out), dimension(1:m) :: C
INTEGER*4 :: i
INTEGER*4 :: j
do i = 1, m
C(i) = 0
end do
do i = 1, m
do j = 1, n
C(i) = B(j)*A(i, j) + C(i)
end do
end do
end subroutine
本文收集自互联网，转载请注明来源。
如有侵权，请联系[email protected] 删除。
我来说两句