pytorch / tensordict

TensorDict is a pytorch dedicated tensor container.
MIT License
832 stars 74 forks source link

Add `_check_dim_name` to `tensorclass` #885

Closed kurtamohler closed 3 months ago

kurtamohler commented 3 months ago

Description

Add _check_dim_name method to tensorclass to prevent warning that the fallback method is not compilable.

Motivation and Context

close #852

Types of changes

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

Checklist

Go over all the following points, and put an x in all the boxes that apply. If you are unsure about any of these, don't hesitate to ask. We are here to help!

vmoens commented 3 months ago

Solved in #896 sorry about that!