input and output dimensions varying while using all in one block.

[krishnarld]

Hi,

Thank you for sharing this repo. I am experiencing an issue while using all in one block. I have relatively small batch of data. My input goes into the block as (6,16). but my output from the block is (2,6). I am not sure why the output dimensions are varying. I tried debugging it myself but I am not quite figure it out.

Can you please help me with this?

Thank you in advance.

[psorrenson]

Hi, this is not the intended behavior. Could you please post a complete example that reproduces this issue so we can look into it? Thanks.