More robust slicing logic for nn.ModuleList (#58361)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/58361
Fixes: https://github.com/pytorch/pytorch/issues/16123
Test Plan: Imported from OSS
Reviewed By: ppwwyyxx
Differential Revision: D28464855
Pulled By: tugsbayasgalan
fbshipit-source-id: db8c41b15dbe6550035e8230dea68ce60e5a6f9a
