打印模型概率

import matplotlib.pyplot as plt
import os

 if not os.path.exists("temp.png"):
        temp = t_prob[3].cpu().detach()
        plt.plot(temp)
        plt.savefig("temp.png")
        plt.close()

其中temp是一维矩阵
结果如下图所示