code: def swap_and_flatten(arr):
      """
      swap and then flatten axes 0 and 1
  
      :param arr: (np.ndarray)
      :return: (np.ndarray)
      """
      shape = arr.shape
      return arr.swapaxes(0, 1).reshape(shape[0] * shape[1], *shape[2:])
  
imports: import time
  import gym
  import numpy
  import tensorflow
  from stable_baselines import logger
  from stable_baselines.common import explained_variance, ActorCriticRLModel, tf_util, SetVerbosity, TensorboardWriter
  from stable_baselines.common.runners import AbstractEnvRunner
  from stable_baselines.common.policies import ActorCriticPolicy, RecurrentActorCriticPolicy
  from stable_baselines.common.schedules import get_schedule_fn
  from stable_baselines.common.tf_util import total_episode_reward_logger
  from stable_baselines.common.math_util import safe_mean
label_desc: Use image node_id 1 for calling a global function with example usage: swap_and_flatten(arr) and returns: arr