Solved by satrajit-chatterjee in post #5 
              
              
                Alright! It worked. Seems like I just needed to pass all the params: 
nn::Functional(max_pool2d, 2, 2, 1, 1, false)

Thanks a lot for all your help!!