Thanks a lot, sir for the reply. I am thinking how did you find this? And also I am just implementing all these to get myself to learn some stuff. I am planning to implement things - if you have any suggestion you can let me know here.
Also what can I do to understand pytorch better?