tensortree (pypi): Represent trees with tensors and perform fast calculations and modifications.
Keywords: tree|tensor|pytorch|numpy|tensortree