我们可以看到绘制出来的图有四个边框，我们通过gca()对坐标轴进行一些简单处理，代码如下。
import torch import torch.nn.functional as F import matplotlib.pyplot as plt import numpy as np x = torch.linspace(-10,10,60) y = torch.sigmoid(x) ax = plt.gca() ax.spines[‘top‘].set_color(‘none‘) #将最上方的边框颜色置为none ax.spines[‘right‘].set_color(‘none‘) #将右边的边框颜色置为none ax.xaxis.set_ticks_position(‘bottom‘) #要移动底部x轴，所以先要锁定x轴 ax.spines[‘bottom‘].set_position((‘data‘,0)) # ‘data‘表示按数值挪动，其后数字代表挪动到Y轴的刻度值 ax.yaxis.set_ticks_position(‘left‘) #同上 ax.spines[‘left‘].set_position((‘data‘,0)) #同上 plt.plot(x.numpy(),y.numpy()) plt.show()
# 要挪动底部的X轴，所以先目光锁定底部！