diff options
Diffstat (limited to 'cumatrix_example.lua')
| -rw-r--r-- | cumatrix_example.lua | 8 |
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) |