Most memory-efficient way to compute abs()**2 of complex numpy ndarray Most memory-efficient way to compute abs()**2 of complex numpy ndarray python python