我使用safetensors提取了我的模型。在原始模型中，我使用了calculate_probabilities = True。当我加载模型时，我想使用transform方法预测主题及其概率。返回的元组只是主题和概率。有没有办法我可以提取所有主题的概率？我的目标是获取文档的前10个主题。