aboutsummaryrefslogtreecommitdiff
path: root/cumatrix_example.lua
diff options
context:
space:
mode:
Diffstat (limited to 'cumatrix_example.lua')
-rw-r--r--cumatrix_example.lua8
1 files changed, 6 insertions, 2 deletions
diff --git a/cumatrix_example.lua b/cumatrix_example.lua
index f26d7f4..aeb5828 100644
--- a/cumatrix_example.lua
+++ b/cumatrix_example.lua
@@ -11,6 +11,10 @@ for i = 0, m - 1 do
end
end
print(fm)
-print(fm:softmax())
+fs = fm:softmax()
+print(fs)
print(dm)
-print(dm:softmax())
+ds = dm:softmax()
+print(ds)
+print(fs + fs)
+print(fs - fs)