Updated function documentation

Fixes #32
Replace totem._isTensor with torch.isTensor