This commit is contained in:
Morvan Zhou
2017-05-13 18:37:25 +10:00
parent e95d798e68
commit 30c0070497

View File

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