Pytorch笔记

Python笔记

最近在学习pytorch，第一个用法是画出激活函数的图像，有些不懂的慢慢记吧
== plt.subplot(xyz) 和其他的基本用法==
表示创建x行y列的画板，并把图片显示在第x行的第z个区域。

y_relu = torch.relu(x).data.numpy()

#创建第一个图像，
plt.figure(1, figsize=(8, 6))

#将第一个画板划分为2行1列组成的区块，并获取到第2块区域
plt.subplot(212)

#plt.plot(x,y,format_string,**kwargs)：x轴数据，y轴数据，format_string控制曲线的格式字串 
format_string 由颜色字符，风格字符，和标记字符 
plt.plot(x_np, y_relu, c='red', label='relu')

#y轴的取值范围
plt.ylim((-1, 5))

#设置坐标轴的lable
plt.xlabel('X axis')
plt.ylabel('Y axis')

#在使用matplotlib画图时，少不了对性能图形做出一些说明和补充。一般情况下，loc属性设置为'best'就足够应付了
plt.legend(loc='best')

#显示图片
plt.show()