Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

关于 RNN 输入形状的问题 #29

Closed
imhuay opened this issue Jan 15, 2018 · 1 comment
Closed

关于 RNN 输入形状的问题 #29

imhuay opened this issue Jan 15, 2018 · 1 comment

Comments

@imhuay
Copy link
Contributor

imhuay commented Jan 15, 2018

我在最新的官方文档上看到 RNN 的输入形状是这样的 [time_steps, batch_size, num_features],它的 batch 在第二维,因为同时给出了一个例子,应该不是输入错误;
然后我看到 TL 的文档里输入是 [batch_size, n_steps, n_features]
是TL在内部重新做了处理吗,还是两种都可以?

@imhuay imhuay changed the title 关于 RNN 输入维度的问题 关于 RNN 输入形状的问题 Jan 15, 2018
@imhuay
Copy link
Contributor Author

imhuay commented Jan 16, 2018

OK,我知道是怎么处理的了
(cell_output, state) = cell(self.inputs[:, time_step, :], state)

@imhuay imhuay closed this as completed Jan 16, 2018
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant