Commit 9626a06

authored

Merge pull request MorvanZhou#66 from jingege315/patch-1

循环的最后一个元素和下一次循环的第一个元素时间不连续

2 parents 51c6c66 + 7f0e442 commit 9626a06Copy full SHA for 9626a06

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -77,7 +77,7 @@ def forward(self, x, h_state):`
`77`	`77`	`for step in range(100):`
`78`	`78`	`start, end = step * np.pi, (step+1)*np.pi # time range`
`79`	`79`	`# use sin predicts cos`
`80`		`- steps = np.linspace(start, end, TIME_STEP, dtype=np.float32) # float32 for converting torch FloatTensor`
	`80`	`+ steps = np.linspace(start, end, TIME_STEP, dtype=np.float32, endpoint=False) # float32 for converting torch FloatTensor`
`81`	`81`	`x_np = np.sin(steps)`
`82`	`82`	`y_np = np.cos(steps)`
`83`	`83`

Comments

(0)