欢迎来到 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
