change softmax to in-place operation and fix memory leak

1 files changed, 4 insertions, 2 deletions

diff --git a/examples/cumatrix_example.lua b/examples/cumatrix_example.lua
index f8235eb..084dcca 100644
--- a/examples/cumatrix_example.lua
+++ b/examples/cumatrix_example.lua
@@ -11,10 +11,12 @@ for i = 0, m - 1 do
     end
 end
 print(fm)
-fs = fm:softmax()
+fs = fm:create()
+fs:softmax(fm)
 -- print(fs)
 print(dm)
-ds = dm:softmax()
+ds = dm:create()
+ds:softmax(dm)
 -- print(ds)
 print(fs)
 print(fs + fs)