Installation

DI-treetensor is currently hosted on PyPI. It required python >= 3.6.

You can simply install DI-treetensor from PyPI with the following command:

pip install di-treetensor

You can also install with the newest version through GitHub:

pip install -U git+https://github.com/opendilab/DI-treetensor.git@main

After installation, run this python code, and version information of treetensor should be shown.

1
2
3
4
5
6
from treetensor.config.meta import __TITLE__, __VERSION__, __AUTHOR__, __DESCRIPTION__

if __name__ == '__main__':
    print(__TITLE__, __VERSION__)
    print(__DESCRIPTION__)
    print('Powered by', __AUTHOR__)

1
2
3
DI-treetensor 0.2.1
A flexible, generalized tree-based tensor structure.
Powered by HansBug, DI-engine's Contributors

DI-treetensor is still under development, you can also check out the documents in stable version through https://opendilab.github.io/DI-treetensor/.