title: Normalize function
body: <p>To normalize, would this work?:</p>
  
  <pre><code>def normalize(arr, axis):
      x = [i for i in range(len(arr.shape))]
      a = x[0]
      x[0] = x[axis]
      x[axis] = a
      arr = arr.transpose(x)
      return (arr/np.sum(arr**2, axis=0)**0.5).transpose(tuple(x))
  </code></pre>