update data_iter (#87)

d2l-ai · Oct 25, 2017 · a06ac2d · a06ac2d
1 parent 3dd61b2
commit a06ac2d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/chapter_recurrent-neural-networks/rnn-scratch.md b/chapter_recurrent-neural-networks/rnn-scratch.md
@@ -80,6 +80,7 @@ def data_iter(batch_size, seq_len, ctx=None):
         return time_numerical[pos:pos+seq_len]
     for i in range(num_batches):
         # 每次读取batch_size个随机样本
+        i = i * batch_size
         examples = idx[i:i+batch_size]
         data = nd.array(
             [_data(j*seq_len) for j in examples], ctx=ctx)