Method flatten

roach/models/ppo_buffer.py:188–193

(arr: np.ndarray)

    @staticmethod
    def flatten(arr: np.ndarray) -> np.ndarray:
        shape = arr.shape
        # if len(shape) < 3:
        #     return arr.swapaxes(0, 1).reshape(shape[0] * shape[1])
        # else:
        return arr.reshape(shape[0] * shape[1], *shape[2:])

    def render(self):
        assert self.full, ''