欢迎来到TensorDict文档！¶
TensorDict 是一个类似字典的类，继承了张量的属性， 如索引、形状操作、转换到设备等。
您可以直接从 PyPI 安装 tensordict（有关安装说明的更多信息，请参见下方专门的章节）：
$ pip install tensordict
TensorDict 的主要目的是通过抽象特定的操作，使代码库更加可读和模块化
>>> for i, tensordict in enumerate(dataset):
... # the model reads and writes tensordicts
... tensordict = model(tensordict)
... loss = loss_module(tensordict)
... loss.backward()
... optimizer.step()
... optimizer.zero_grad()
借助这一抽象层级，用户可复用训练循环来处理高度异构的任务。 训练循环中的每个独立步骤（如数据采集与变换、模型预测、损失计算等） 均可根据当前具体应用场景进行定制，而不会影响其他步骤。 例如，上述示例可轻松应用于分类、分割等多种任务。
安装¶
Tensordict 的版本发布与 PyTorch 同步，因此请务必使用最新版本的 PyTorch，以始终享受该库的最新功能（尽管核心功能保证向后兼容 PyTorch >=1.13）。 可通过以下方式安装夜间构建版本：
$ pip install tensordict-nightly
或通过 git clone 如果你愿意为库做出贡献：
$ cd path/to/root
$ git clone https://github.com/pytorch/tensordict
$ cd tensordict
$ python setup.py develop