• Docs >
  • Welcome to the TensorDict Documentation!
Shortcuts

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

教程

基础

tensordict.nn

数据加载

目录

索引和表格