code: def batch_rms_2d(array: np.ndarray) -> np.ndarray:
      return batch_rms(
          array.reshape(array.shape[0], array.shape[1] * array.shape[2])
      )
  
imports: import numpy
  import statsmodels.api
label_desc: Use image node_id 21 for calling a global function with example usage: batch_rms_2d(array) and returns: batch_rms