def METHOD_NAME(self, encoder_out, new_order):
    encoder_out["encoder_out"] = encoder_out["encoder_out"].index_select(
        1, new_order
    )
    encoder_out["encoder_padding_mask"] = encoder_out[
        "encoder_padding_mask"
    ].index_select(1, new_order)
    return encoder_out