m

core.py

@@ -155,11 +155,11 @@ class omega_op(gof.PythonOp):
 
 
 def scalar_switch(x, y, normal_f, scalar_f):
-    x, y = wrap(x), wrap(y)
-    if x.constant and not x.data.shape:
-        return scalar_f(y, x)
-    if y.constant and not y.data.shape:
-        return scalar_f(x, y)
+#     x, y = wrap(x), wrap(y)
+#     if x.constant and not x.data.shape:
+#         return scalar_f(y, x)
+#     if y.constant and not y.data.shape:
+#         return scalar_f(x, y)
     return normal_f(x, y)
 
 
@@ -231,10 +231,13 @@ class iadd(proto_add, inplace):
 class add_scalar(omega_op):
     impl = numpy.ndarray.__add__
 
+class iadd_scalar(omega_op, inplace):
+    impl = numpy.ndarray.__add__
+
 
 class proto_twice(omega_op):
     def grad(x, gz):
-        return scal(gz, 2.0)
+        return scale(gz, 2.0)
 
 class twice(proto_twice):
     def impl(x):
@@ -280,7 +283,7 @@ class imul(proto_mul, inplace):
 
 class proto_sqr(omega_op):
     def grad(x, gz):
-        return scal(mul(x, gz), 2.0)
+        return scale(mul(x, gz), 2.0)
 
 class sqr(proto_sqr):
     impl = lambda x: numpy.multiply(x, x)
@@ -291,7 +294,7 @@ class isqr(proto_sqr, inplace):
 
 class proto_sqrt(omega_op):
     def grad(x, gz):
-        return scal(div(gz, sqrt(x)), 0.5)
+        return scale(div(gz, sqrt(x)), 0.5)
 
 class sqrt(proto_sqrt):
     impl = numpy.sqrt
@@ -329,7 +332,7 @@ class iinv_scale(omega_op, inplace):
 
 class proto_scale(omega_op):
     def grad(x, a, gz):
-        return scal(a, gz), sum(mul(x, gz))
+        return scale(a, gz), sum(mul(x, gz))
 
 class scale(omega_op):
     impl = numpy.ndarray.__mul__

test.py

@@ -131,6 +131,7 @@ import grad
 
 ############################
 
+
 # def dataset_1hot(x, targ, n):
 #     """Return an looping iterator over 1-hot vectors
 
@@ -181,15 +182,6 @@ import grad
 
 
 
-# # 1 = mul(mul(neg(scal(mul(sub(0.736213102665, sigmoid(*3)), 1.0), 2.0)), sigmoid(*3)), sub(1, sigmoid(*3)))
-# # 2 = transpose(0.11474051836)
-# # 3 = dot(*2, *5)
-# # 4 = dot(0.11474051836, 0.736213102665)
-# # 5 = sigmoid(*4)
-# # add(transpose(dot(*1, transpose(*5))), dot(mul(mul(dot(transpose(*2), *1), sigmoid(*4)), sub(1, sigmoid(*4))), transpose(0.736213102665)))
-
-
-
 
 
 ############################
@@ -209,8 +201,31 @@ import grad
 
 ############################
 
-print core.ones((2, 2)) + 1
+# print core.ones((2, 2)) + 1
+
+# print numpy.ones((2, 2)) ** numpy.ones((2, 2))
+
+
+############################
+
+
+x = core.ones((2, 2))
+y = core.zeros((1, 1))
+
+#print "?", gof.graph.ops([], [x + y])
+
+
+print x
+
+x + x
+print "1", gof.eval_env#.ops()
+y + y
+print "2", gof.eval_env#.ops()
+x + x
+print "3", gof.eval_env#.ops()
+
 
-print numpy.ones((2, 2)) ** numpy.ones((2, 2))
+x += (x + x)
+print x