file theano/misc/may_share_memory.py

theano/misc/may_share_memory.py

@@ -4,7 +4,7 @@ numpy version support only ndarray.
 """
 from __future__ import absolute_import, print_function, division
 
-import numpy
+import numpy as np
 from theano.tensor.basic import TensorType
 
 try:
@@ -42,8 +42,8 @@ else:
 
 
 def may_share_memory(a, b, raise_other_type=True):
-    a_ndarray = isinstance(a, numpy.ndarray)
-    b_ndarray = isinstance(b, numpy.ndarray)
+    a_ndarray = isinstance(a, np.ndarray)
+    b_ndarray = isinstance(b, np.ndarray)
     if a_ndarray and b_ndarray:
         return TensorType.may_share_memory(a, b)
     a_cuda = _is_cuda(a)