def METHOD_NAME(key):
    if key.size(2) < self.pre_decision_ratio:
        return key
    else:
        k = key[
            :,
            :,
            self.pre_decision_ratio - 1:: self.pre_decision_ratio,
        ].contiguous()
        if key.size(-1) % self.pre_decision_ratio != 0:
            k = torch.cat([k, key[:, :, -1:]], dim=-1).contiguous()
        return k