Added doctests in tensor_free_module.py