docs: def normalise_tensor(tensor):
      '''
      Normalise the tensor by dividing it by its norm, defined such that
      np.sqrt(X:X)
      '''
      tensor_norm = np.linalg.norm(tensor)
      return tensor / tensor_norm, tensor_norm