file theano/gpuarray/neighbours.py

theano/gpuarray/neighbours.py

 from __future__ import absolute_import, print_function, division
-import numpy
+import numpy as np
 
 from theano import Op, Apply, config
 from theano.tensor.nnet.neighbours import Images2Neibs
@@ -256,8 +256,8 @@ class GpuImages2Neibs(GpuKernelBase, Images2Neibs, Op):
         dtype_neib_shape = node.inputs[1].dtype
         dtype_neib_step = node.inputs[2].dtype
         dtype_z = node.outputs[0].dtype
-        itemsize_ten4 = numpy.dtype(dtype_ten4).itemsize
-        itemsize_z = numpy.dtype(dtype_z).itemsize
+        itemsize_ten4 = np.dtype(dtype_ten4).itemsize
+        itemsize_z = np.dtype(dtype_z).itemsize
         typecode_z = pygpu.gpuarray.dtype_to_typecode(node.outputs[0].dtype)
         ten4, neib_shape, neib_step = inp
         z, = out