[Numpy-discussion] Optimized sum of squares

Gael Varoquaux gael.varoquaux at normalesup.org
Sun Oct 18 08:09:27 EDT 2009


On Sun, Oct 18, 2009 at 09:06:15PM +1100, Gary Ruben wrote:
> 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.

Ha, yes. Good point about the tensordot trick.

Thank you

Gaël



More information about the NumPy-Discussion mailing list