Add zeros, ones, randn to funsor.compat.ops (#270)

pyro-ppl · Oct 11, 2019 · 7810696 · 7810696
1 parent f115484
commit 7810696
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/funsor/compat/__init__.py b/funsor/compat/__init__.py
diff --git a/funsor/compat/ops.py 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
@@ -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',
 ]
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,3 @@
		from torch import ones, randn, tensor, zeros # noqa F401

		from .ops import * # noqa F401