欢迎使用 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）。 Nightly 版本可以通过以下方式安装
$ pip install tensordict-nightly
或者如果你愿意为库做出贡献，也可以通过 git clone 来调用：
$ cd path/to/root
$ git clone https://github.com/pytorch/tensordict
$ cd tensordict
$ python setup.py develop