我是一个初学者，请你帮我解释以下这段代码的每个变量的含义和每句代码的语法：def top_k_top_p_filtering(logits, top_k=0, top_p=0.0, filter_value=-float('Inf')): assert logits.dim() == 1 # batch size 1 for now - could be updated for more but the code would be less clear top_k = min(top_k, logits.size(-1)) # Safety check if top_k > 0: # Remove all tokens with a probability less than the last token of the top-k # torch.topk()返回最后一维最大的top_k个元素，返回值为二维(values,indices) # ...表示其他维度由计算机自行推断 indices_to_remove = logits < torch.topk(logits, top_k)[0][..., -1, None] logits[indices_to_remove] = filter_value if top_p > 0.0: sorted_logits, sorted_indices = torch.sort(logits, descending=True) cumulative_probs = torch.cumsum(F.softmax(sorted_logits, dim=-1), dim=-1) # Remove tokens with cumulative probability above the threshold sorted_indices_to_remove = cumulative_probs > top_p # Shift the indices to the right to keep also the first token above the threshold sorted_indices_to_remove[..., 1:] = sorted_indices_to_remove[..., :-1].clone() sorted_indices_to_remove[..., 0] = 0 indices_to_remove = sorted_indices[sorted_indices_to_remove] logits[indices_to_remove] = filter_value return logits

这段代码的作用是根据给定的top_k和top_p值，过滤掉logits分数较低的预测结果，从而生成更准确的预测结果。下面是每个变量的含义和每句代码的语法：

1. `logits`：一个张量，表示对应词汇表中单词的logits分数。

2. `top_k`：一个整数，表示要保留的最高可能性预测的数量。

3. `top_p`：一个浮点数，表示要保留的累积概率质量。

4. `filter_value`：一个浮点数，用于过滤掉不想要的预测。默认值为负无穷大。

5. `assert`：断言语句，用于判断logits张量的维度是否为1，如果维度不为1，程序将会报错并停止运行。

6. `logits.dim()`：张量的维度数。

7. `top_k = min(top_k, logits.size(-1))`：将top_k值与logits张量的最后一维大小进行比较，保证top_k值不会大于张量的维度。

8. `if top_k > 0:`：如果指定了top_k值，则进行以下操作。

9. `indices_to_remove = logits < torch.topk(logits, top_k)[0][..., -1, None]`：返回logits张量中最后一维的最大值的top_k个元素，并将剩余元素的值设置为过滤值, 然后返回不需要的结果的索引。

10. `logits[indices_to_remove] = filter_value`：将logits张量中的索引为indices_to_remove的元素的值设置为过滤值。

11. `if top_p > 0.0:`：如果指定了top_p值，则进行以下操作。

12. `sorted_logits, sorted_indices = torch.sort(logits, descending=True)`：按照降序对logits张量进行排序，并返回排序后的结果和对应的索引。

13. `cumulative_probs = torch.cumsum(F.softmax(sorted_logits, dim=-1), dim=-1)`：计算softmax函数的累积概率值。

14. `sorted_indices_to_remove = cumulative_probs > top_p`：返回累积概率大于top_p的索引。

15. `sorted_indices_to_remove[..., 1:] = sorted_indices_to_remove[..., :-1].clone()`：将索引向右移一位，保留第一个索引。

16. `sorted_indices_to_remove[..., 0] = 0`：将第一个索引设置为0。

17. `indices_to_remove = sorted_indices[sorted_indices_to_remove]`：返回不需要的结果的索引。

18. `logits[indices_to_remove] = filter_value`：将logits张量中的索引为indices_to_remove的元素的值设置为过滤值。

19. `return logits`：返回过滤后的logits张量。