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是一维矩阵
结果如下图所示
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是一维矩阵
结果如下图所示