[Numpy-discussion] Optimized sum of squares

Gary Ruben gruben at bigpond.net.au
Sun Oct 18 06:06:15 EDT 2009


Hi Gaël,

If you've got a 1D array/vector called "a", I think the normal idiom is

np.dot(a,a)

For the more general case, I think
np.tensordot(a, a, axes=something_else)
should do it, where you should be able to figure out something_else for 
your particular case.

Gary R.

Gael Varoquaux wrote:
> On Sat, Oct 17, 2009 at 07:27:55PM -0400, josef.pktd at gmail.com wrote:
>>>>> Why aren't you using logaddexp ufunc from numpy?
> 
>>>> Maybe because it is difficult to find, it doesn't have its own docs entry.
> 
> Speaking of which...
> 
> I thought that there was a readily-written, optimized function (or ufunc)
> in numpy or scipy that calculated the sum of squares for an array
> (possibly along an axis). However, I cannot find it.
> 
> Is there something similar? If not, it is not the end of the world, the
> operation is trivial to write.
> 
> Cheers,
> 
> Gaël



More information about the NumPy-Discussion mailing list