I am having a hard time with tensorflow lately where it obliterates shape information when I do something like batch_flatten.  I find this very annoying as its a lot of work to verify my calculations are functioning as they should instead of by some luck through broadcasting.
Has anyone experienced this in tf and know what i am talking about?  Does pytorch improve up on this?  If the answer is yes, this is reason enough to switch.
Thank you,
Isaac