class_docstr: Transpose observations.
class_docstr_tok: ["Transpose","observations","."]
code: def __init__(self, env, axes):
      super().__init__(env)
      self._axes = axes
      assert isinstance(env.observation_space, gym.spaces.Box)
      self.observation_space = gym.spaces.Box(
          low=env.observation_space.low.transpose(*self._axes),
          high=env.observation_space.high.transpose(*self._axes),
          dtype=env.observation_space.dtype,
      )
  
imports: import argparse
  import functools
  import os
  import gym
  import gym.spaces
  import numpy
  import torch
  from torch import nn
  import pfrl
  from pfrl import experiments, explorers, replay_buffers, utils
  from pfrl.q_functions import DiscreteActionValueHead
label_desc: Use image node_id 1 to create a new TransposeObservation object from inherited base classes: gym with example: obj = TransposeObservation(env, axes)