Merge pull request #769 from larseeri/shared_borrow

theano/compile/sharedvalue.py

@@ -4,6 +4,7 @@ __docformat__ = 'restructuredtext en'
 # Standard imports
 import copy
 import logging
+import sys
 
 # Third-party imports
 import numpy
@@ -179,6 +180,9 @@ def shared(value, name=None, strict=False, allow_downcast=None, **kwargs):
            `See <http://deeplearning.net/software/theano/tutorial/aliasing.html#borrowing-when-creating-shared-variables>`_ for detail.
 
     """
+
+    try:
+
         for ctor in reversed(shared.constructors):
             try:
                 return ctor(value, name=name, strict=strict,
@@ -190,8 +194,15 @@ def shared(value, name=None, strict=False, allow_downcast=None, **kwargs):
             #
             # This was done on purpose, the rationale being that if kwargs
             # were supplied, the user didn't want them to be ignored.
+
+    except MemoryError, e:
+        e.args = e.args + ('you might consider'
+                           ' using \'theano.shared(..., borrow=True)\'',)
+        raise
+
     raise TypeError('No suitable SharedVariable constructor could be found',
                     (value, kwargs))
+
 shared.constructors = []