diff --git a/funsor/compat/__init__.py b/funsor/compat/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/funsor/compat/ops.py b/funsor/compat/ops.py new file mode 100644 index 000000000..60161922c --- /dev/null +++ b/funsor/compat/ops.py @@ -0,0 +1,3 @@ +from torch import ones, randn, tensor, zeros # noqa F401 + +from .ops import * # noqa F401 diff --git a/funsor/ops.py b/funsor/ops.py index 38946be9f..66c5a81aa 100644 --- a/funsor/ops.py +++ b/funsor/ops.py @@ -3,7 +3,6 @@ import numpy as np from multipledispatch import Dispatcher -from torch import tensor _builtin_abs = abs _builtin_max = max @@ -345,7 +344,6 @@ def reciprocal(x): 'sigmoid', 'sqrt', 'sub', - 'tensor', 'truediv', 'xor', ]