conver data_iter() function to tensorflow version

[Skyexu]

修改 date_iter() 方法，原来是将 tensor 转换为 numpy 进行处理的，改为直接使用 tensor，使用 tf.gather 进行切片。 code 和 docs 里面相应的内容都修改了一下。

[archersama]

如果可以代码运行无误，可以合并

[Skyexu]

直接使用tensor，避免转换开销，和原书也契合一些。本节的代码在我本地都试过没问题😀。

[SwordFaith]

agree