python平方和psum_numpy / scipy中的平方差异（SSD）总和

I'm trying to use Python and Numpy/Scipy to implement an image processing algorithm. The profiler tells me a lot of time is being spent in the following function (called often), which tells me the sum of square differences between two images

def ssd(A,B):

s = 0

for i in range(3):

s += sum(pow(A[:,:,i] - B[:,:,i],2))

return s

How can I speed this up? Thanks.

解决方案

Just

s = numpy.sum((A[:,:,0:3]-B[:,:,0:3])**2)

(which I expect is likely just sum((A-B)**2) if the shape is always (,,3))

You can also use the sum method: ((A-B)**2).sum()

Right?