    def get_output_dim(self):
        """Returns the dimension of the output.

        Returns: The dimension of the output.
        """
        return self.output_dim
Returns the dimension of the output.

Returns: The dimension of the output.
mbpo/models/fc.py