Results 3 comments of Nikita Rath

For the error in the second case you have mentioned - ```RuntimeError: view size is not compatible with input tensor's size and stride (at least one dimension spans across two...

Hi, I can provide code for this problem. Could you please assign this issue to me?

Ok got it. Thanks for the info