Python函数:返回一个没有额外内存分配的数组
Python函数:返回一个没有额外内存分配的数组
提问于 2018-07-26 04:28:13
假设我想做一个用输入向量乘以输入矩阵的函数：
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。
谢谢,
米哈伊尔
回答 2
Stack Overflow用户
回答已采纳
发布于 2018-07-26 04:35:20
Numpy在它的大多数生成新数组的函数中，都有一个参数来存储结果数组。我在下面的代码中使用了该参数的名称版本out，但您可以省略该名称。您必须确保out数组具有正确的形状和数据类型。这个参数的目的正是您想要的--避免额外的内存分配。这也可以加快代码的速度。
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')
A.dot(b, out=c)documentation for dot()中提到了该参数。如果您想要更改函数MatMul的定义，
def MatMul(A, b, c=None):
return A.dot(b, out=c)并将调用更改为
MatMul(A, b, c)Stack Overflow用户
发布于 2018-07-26 04:34:33
dot不知道也不关心c变量或该变量已经包含引用的数组。它将生成一个新的数组，=会将c变量绑定到这个新的数组，让内存管理系统清理旧的数组。
如果您希望dot将输出存储到现有数组中，则需要告诉它这样做：
A.dot(b, out=c)页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/51527164
复制相关文章
相似问题
腾讯云开发者
