def METHOD_NAME(self, incremental_state, new_order):
    input_buffer = self._get_input_buffer(incremental_state)
    if input_buffer is not None:
        input_buffer = input_buffer.index_select(1, new_order)
        self._set_input_buffer(incremental_state, input_buffer)