    def set_output_dim(self, output_dim):
        """Sets the dimension of the output.

        Arguments:
            output_dim: (int) The dimension of the output.

        Returns: None.
        """
        if self.variables_constructed:
            raise RuntimeError("Variables already constructed.")
        self.output_dim = output_dim
Sets the dimension of the output.

Arguments:
output_dim: (int) The dimension of the output.

Returns: None.
mbpo/models/fc.py