add exp, log for ndarray

halisyilboga1 · Oct 27, 2015 · 87c2492 · 87c2492
1 parent db1a5b5
commit 87c2492
Show file tree

Hide file tree

Showing 2 changed files with 37 additions and 0 deletions.
diff --git a/src/ndarray/unary_function-inl.h b/src/ndarray/unary_function-inl.h
@@ -107,6 +107,18 @@ MXNET_REGISTER_TBLOB_FUN(sqrt, XPU)
 .set_gradient(XPU::kDevMask, UnaryBackwardUseOut_<XPU, square_root_grad>, true)
 .describe("Take square root of the src");
 
+// exp
+MXNET_REGISTER_TBLOB_FUN(exp, XPU)
+.set_function(XPU::kDevMask, UnaryForward_<XPU, op::mshadow_op::exp>, true)
+.set_gradient(XPU::kDevMask, UnaryBackwardUseOut_<XPU, op::mshadow_op::exp_grad>, true)
+.describe("Take exp of the src");
+
+//log
+MXNET_REGISTER_TBLOB_FUN(log, XPU)
+.set_function(XPU::kDevMask, UnaryForward_<XPU, op::mshadow_op::log>, true)
+.set_gradient(XPU::kDevMask, UnaryBackwardUseOut_<XPU, op::mshadow_op::log_grad>, true)
+.describe("Take log of the src");
+
 // L2 norm
 MXNET_REGISTER_TBLOB_FUN(norm, XPU)
 .set_function(XPU::kDevMask, L2Norm<XPU>, false, false)

diff --git a/src/operator/mshadow_op.h b/src/operator/mshadow_op.h
@@ -80,6 +80,31 @@ struct tanh_grad {
   }
 };
 
+
+struct exp {
+  MSHADOW_XINLINE static real_t Map(real_t a) {
+    return expf(a);
+  }
+};
+struct exp_grad {
+  MSHADOW_XINLINE static real_t Map(real_t a) {
+    return expf(a);
+  }
+};
+
+struct log {
+  MSHADOW_XINLINE static real_t Map(real_t a) {
+    return logf(a);
+  }
+};
+struct log_grad	{
+  MSHADOW_XINLINE static real_t Map(real_t a) {
+    return 1.0f/a;
+  }
+};
+
+
+
 struct square {
   MSHADOW_XINLINE static real_t Map(real_t a) {
     return a * a;