假设我想做一个用输入向量乘以输入矩阵的函数:
def MatMul(A,b):
return A.dot(b)
现在,我执行以下代码:
import numpy as np
A=np.array([[1,2,3],[4,5,6],[7,8,9]],dtype='float64')
b=np.array([4,5,6],dtype='float64')
c=np.zeros(3,dtype='float64')
c=MatMul(A,b)
在MatMul
函数中会有额外的数组分配吗?我知道A
和b
将通过引用传递。注意,我已经预先分配了数组c
。
通常,如何在这样的简单函数中避免不必要的预分配?比如说,我想执行几个数学运算:
def Rank1Update(A,b,alpha):
c=A.dot(b)
c+=alpha*c.dot(c)*c
return c
我可以在一行中包含很多数学函数,但是代码很快就变得不可读了。
我熟悉C风格的编程,在这种编程中,为了避免不必要的内存分配,可以通过引用传递A
、b
和c
,并在返回void
的函数中更新c
。我可以在python中做同样的事情,但为了方便和代码可读性,我想使用return
。
谢谢,
米哈伊尔
https://stackoverflow.com/questions/51527164
复制相似问题