little fixes

上级 bc361363
...@@ -231,6 +231,9 @@ class iadd(proto_add, inplace): ...@@ -231,6 +231,9 @@ class iadd(proto_add, inplace):
class add_scalar(omega_op): class add_scalar(omega_op):
impl = numpy.ndarray.__add__ impl = numpy.ndarray.__add__
class iadd_scalar(omega_op):
impl = numpy.ndarray.__iadd__
class proto_twice(omega_op): class proto_twice(omega_op):
def grad(x, gz): def grad(x, gz):
...@@ -280,7 +283,7 @@ class imul(proto_mul, inplace): ...@@ -280,7 +283,7 @@ class imul(proto_mul, inplace):
class proto_sqr(omega_op): class proto_sqr(omega_op):
def grad(x, gz): def grad(x, gz):
return scal(mul(x, gz), 2.0) return scale(mul(x, gz), 2.0)
class sqr(proto_sqr): class sqr(proto_sqr):
impl = lambda x: numpy.multiply(x, x) impl = lambda x: numpy.multiply(x, x)
...@@ -291,7 +294,7 @@ class isqr(proto_sqr, inplace): ...@@ -291,7 +294,7 @@ class isqr(proto_sqr, inplace):
class proto_sqrt(omega_op): class proto_sqrt(omega_op):
def grad(x, gz): def grad(x, gz):
return scal(div(gz, sqrt(x)), 0.5) return scale(div(gz, sqrt(x)), 0.5)
class sqrt(proto_sqrt): class sqrt(proto_sqrt):
impl = numpy.sqrt impl = numpy.sqrt
...@@ -329,7 +332,7 @@ class iinv_scale(omega_op, inplace): ...@@ -329,7 +332,7 @@ class iinv_scale(omega_op, inplace):
class proto_scale(omega_op): class proto_scale(omega_op):
def grad(x, a, gz): 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): class scale(omega_op):
impl = numpy.ndarray.__mul__ impl = numpy.ndarray.__mul__
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论