The input feature dimension in last batch is 1 while in other batches is 2 !!!!

[moghadas76]

[chaoshangcs]

Thanks for you questions. I have debugged our code about this question but haven't found it. If you can provide more details, that will be great. Thanks!