|
20 | 20 | LR = 0.02 # learning rate |
21 | 21 |
|
22 | 22 | # show data |
23 | | -steps = np.linspace(0, np.pi*2, 100, dtype=np.float32) |
24 | | -x_np = np.sin(steps)# float32 for converting torch FloatTensor |
| 23 | +steps = np.linspace(0, np.pi*2, 100, dtype=np.float32)# float32 for converting torch FloatTensor |
| 24 | +x_np = np.sin(steps) |
25 | 25 | y_np = np.cos(steps) |
26 | 26 | plt.plot(steps, y_np, 'r-', label='target (cos)') |
27 | 27 | plt.plot(steps, x_np, 'b-', label='input (sin)') |
@@ -71,8 +71,8 @@ def forward(self, x, h_state): |
71 | 71 | for step in range(100): |
72 | 72 | start, end = step * np.pi, (step+1)*np.pi # time range |
73 | 73 | # use sin predicts cos |
74 | | - steps = np.linspace(start, end, TIME_STEP, dtype=np.float32) |
75 | | - x_np = np.sin(steps)# float32 for converting torch FloatTensor |
| 74 | + steps = np.linspace(start, end, TIME_STEP, dtype=np.float32)# float32 for converting torch FloatTensor |
| 75 | + x_np = np.sin(steps) |
76 | 76 | y_np = np.cos(steps) |
77 | 77 |
|
78 | 78 | x = torch.from_numpy(x_np[np.newaxis, :, np.newaxis]) # shape (batch, time_step, input_size) |
|
0 commit comments