opendilab / di-treetensor Goto Github PK

View Code? Open in Web Editor NEW

181.0 3.0 5.0 72.6 MB

Let DI-treetensor help you simplify the structure processing!（树形运算一不小心就逻辑混乱？DI-treetensor快速帮你搞定）

Home Page: https://opendilab.github.io/DI-treetensor/

License: Apache License 2.0

Makefile 0.16% Shell 0.71% Python 99.12%

tensor tree-structure numpy pytorch python deep-learning reinforcement-learning

di-treetensor's Issues

Failure when try to convert between numpy and torch on Windows Python3.10

See here: https://github.com/opendilab/DI-treetensor/runs/7820313811?check_suite_focus=true

The bug is like

    @method_treelize(return_type=_get_tensor_class)
    def tensor(self: numpy.ndarray, *args, **kwargs):
>       tensor_: torch.Tensor = torch.from_numpy(self)
E       RuntimeError: Numpy is not available

The only way I found to 'solve' this is to downgrade python to version3.9 to lower. So these tests will be skipped temporarily.

Special OP List

Reduce Result OP

Input/Output Tuple OP

stack
cat
split

High Precision OP

mean
std
sum

PyTorch OP List(P0)

reference: https://pytorch.org/docs/1.8.0/torch.html

common

numel
cpu
cuda
to

Creation Ops

Indexing, Slicing, Joining, Mutating Ops

Math Ops

Pointwise Ops

Reduction Ops

Comparison Ops

Other Ops

cdist
clone
flip

opendilab / di-treetensor Goto Github PK

di-treetensor's People

Contributors

Stargazers

Watchers

Forkers