使用Python将PyTorch Top-k结果保存到TXT文件

您可以使用Python的numpy库将结果保存到txt文件中。首先，将结果转换为numpy数组，并使用numpy.savetxt()函数将其保存为txt文件。以下是一个示例代码：

import numpy as np

# 将结果转换为numpy数组
top5_probs_arr = top5_probs.detach().numpy()
top5_indice_arr = top5_indice.detach().numpy()

# 将结果保存为txt文件
np.savetxt('top5_probs.txt', top5_probs_arr)
np.savetxt('top5_indice.txt', top5_indice_arr)

上述代码将结果分别保存为top5_probs.txt和top5_indice.txt两个txt文件。