diff misc.py @ 175:e9a95e19e6f8

Added a Print Op
author Yoshua Bengio <bengioy@iro.umontreal.ca>
date Tue, 13 May 2008 15:11:24 -0400
parents 718befdc8671
children d7250ee86f72
line wrap: on
line diff
--- a/misc.py	Tue May 13 14:28:47 2008 -0400
+++ b/misc.py	Tue May 13 15:11:24 2008 -0400
@@ -1,3 +1,24 @@
+
+import theano
+
+class Print(theano.Op):
+    def __init__(self,message=""):
+        self.message=message
+        self.view_map={0:[0]}
+
+    def make_node(self,xin):
+        xout = xin.type.make_result()
+        return theano.Apply(op = self, inputs = [xin], outputs=[xout])
+
+    def perform(self,node,inputs,output_storage):
+        xin, = inputs
+        xout, = output_storage
+        xout[0] = xin
+        print self.message,xin
+
+    def grad(self,input,output_gradients):
+        return output_gradients
+
 
 def unique_elements_list_intersection(list1,list2):
     """