Merge pull request #66 from jingege315/patch-1

循环的最后一个元素和下一次循环的第一个元素时间不连续
This commit is contained in:
Morvan
2018-12-03 14:32:44 +08:00
committed by GitHub

View File

@ -77,7 +77,7 @@ plt.ion() # continuously plot
for step in range(100):
start, end = step * np.pi, (step+1)*np.pi # time range
# use sin predicts cos
steps = np.linspace(start, end, TIME_STEP, dtype=np.float32) # float32 for converting torch FloatTensor
steps = np.linspace(start, end, TIME_STEP, dtype=np.float32, endpoint=False) # float32 for converting torch FloatTensor
x_np = np.sin(steps)
y_np = np.cos(steps)