Pass tensordict as an additional input of _precompute_cache

[Junyoungpark]

Description

Pass td as an additional input of _precompute_cache to allow better flexibility of _precompute_cache in the children classes

Motivation and Context

Types of changes

What types of changes does your code introduce? Remove all that do not apply:

• New feature (non-breaking change which adds core functionality)

Checklist

• My change requires a change to the documentation.
• I have updated the tests accordingly (required for a bug fix or a new feature).
• I have updated the documentation accordingly.

[codecov]

Codecov Report

Patch coverage is 85.71% of modified lines.

Files Changed	Coverage
rl4co/models/zoo/common/autoregressive/decoder.py	85.71%

📢 Thoughts on this report? Let us know!.

[Junyoungpark]

@fedebotu would you take a look at this?
I found passing td as an extra argument for _precompute_cache can be useful to design a subclass of AutoregressivePolicy with the extended features.

[fedebotu]

I left a minor comment as review - I think it may be useful to substitute the PrecomputedCache into something more TensorDict-like in the future - for instance, TensorClass may do

[Junyoungpark]

I leave the reply comments in the code!
yeah, I agree with the suggestion!