code: def _partial_flatten_and_normalize(x):
      """Flatten all but the first dimension of an `np.ndarray`."""
      x = np.reshape(x, (x.shape[0], -1))
      return (x - np.mean(x)) / np.std(x)
  
file_path: neural-tangents/examples/datasets.py
imports: import gzip
  import os
  import shutil
  import urllib.request
  from jax import random
  import numpy
  import tensorflow
  import tensorflow_datasets
label_desc: Use image node_id 1 for calling a global function with example usage: _partial_flatten_and_normalize(x) and returns: unknown