    def get_input_dim(self):
        """Returns the dimension of the input.

        Returns: The dimension of the input
        """
        return self.input_dim
Returns the dimension of the input.

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