Pulls the observation from PyTorch and sets up for Matplotlib plotting. :param obs: A 2D array of floats depicting an input image. :param tag: A unique tag to associate the data with. :param step: The step of the pipeline.
(self, obs: torch.Tensor, tag: str = "obs", step: int = None)
| 28 | |
| 29 | @abstractmethod |
| 30 | def plot_obs(self, obs: torch.Tensor, tag: str = "obs", step: int = None) -> None: |
| 31 | # language=rst |
| 32 | """ |
| 33 | Pulls the observation from PyTorch and sets up for Matplotlib |
| 34 | plotting. |
| 35 | |
| 36 | :param obs: A 2D array of floats depicting an input image. |
| 37 | :param tag: A unique tag to associate the data with. |
| 38 | :param step: The step of the pipeline. |
| 39 | """ |
| 40 | |
| 41 | @abstractmethod |
| 42 | def plot_reward( |
no outgoing calls