def _reset_bind_size(self, new_size):
        """reset input shape of the model

        Parameters
        ----------
        new_size: int
            new batch size
        """
        if self.bind_size == new_size:
            return
        else:
            self.bind_size = new_size
            self.model.reshape(
                data_shapes=[
                    ('input_view',    (new_size,) + self.view_space),
                    ('input_feature', (new_size,) + self.feature_space)],
            )
reset input shape of the model

        Parameters
        ----------
        new_size: int
            new batch size
        
python/magent/builtin/mx_model/a2c.py