[功能建议] 支持 tensordict (以及 tensorclass)

Tensordict 是具有共同批量维度的张量字典, tensorclasses 是张量的数据类。 https://pytorch.org/tensordict/ Tensordict 似乎还没有完全成熟,但已经非常有用。如果我没记错的话,它支持写入 einops 后端所需的大多数操作 - einsum 缺失。