treetensor.torch.funcs