[Feature] Add _Contiguous module and reshape improvements to encoders/decoders

[vmoens]

Stack from ghstack (oldest at bottom):

• -> [Feature] Add _Contiguous module and reshape improvements to encoders/decoders #3306
• [BugFix] Fix SliceSampler for torch.compile compatibility #3298

---

• Add _Contiguous helper module for torch.compile inductor compatibility
• Refactor ObsEncoder.forward and ObsDecoder.forward to use flatten/unflatten with contiguous()
• Add _maybe_record_function_decorator for profiling

[pytorch-bot]

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/pytorch/rl/3306

• 📄 Preview Python docs built from this PR

Note: Links to docs will display an error until the docs builds have been completed.

❌ 7 New Failures, 1 Unrelated Failure

As of commit c2236ba with merge base d7ef78b ():

NEW FAILURES - The following jobs have failed:

• Unit-tests on Linux / tests-cpu (3.10) / linux-job (gh)
  test/modules/test_dreamer_components.py::TestDreamerComponents::test_dreamer_decoder_out_channels[20-10-3-64-batch_size0-device0]
• Unit-tests on Linux / tests-cpu (3.11) / linux-job (gh)
  test/modules/test_dreamer_components.py::TestDreamerComponents::test_dreamer_decoder_out_channels[20-10-3-64-batch_size0-device0]
• Unit-tests on Linux / tests-cpu (3.12) / linux-job (gh)
  test/modules/test_dreamer_components.py::TestDreamerComponents::test_dreamer_decoder_out_channels[20-10-3-64-batch_size0-device0]
• Unit-tests on Linux / tests-cpu (3.13) / linux-job (gh)
  test/modules/test_dreamer_components.py::TestDreamerComponents::test_dreamer_decoder_out_channels[20-10-3-64-batch_size0-device0]
• Unit-tests on Linux / tests-cpu (3.14) / linux-job (gh)
  test/modules/test_dreamer_components.py::TestDreamerComponents::test_dreamer_decoder_out_channels[20-10-3-64-batch_size0-device0]
• Unit-tests on Linux / tests-olddeps (3.10, 11.8) / linux-job (gh)
  test/modules/test_dreamer_components.py::TestDreamerComponents::test_dreamer_decoder_out_channels[20-10-3-64-batch_size0-device0]
• Unit-tests on Linux / tests-optdeps (3.12, 13.0) / linux-job (gh)
  test/modules/test_dreamer_components.py::TestDreamerComponents::test_dreamer_decoder_out_channels[20-10-3-64-batch_size0-device0]

FLAKY - The following job failed but was likely due to flakiness present on trunk:

• Unit-tests on Windows / unittests-cpu (3.10, windows.4xlarge, cpu) / windows-job (gh) (matched win rule in flaky-rules.json)
  ##[error]The operation was canceled.

This comment was automatically generated by Dr. CI and updates every 15 minutes.