blob: aeb5828b15729ef4c433e752191b71a48c13df81 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
|
m = 10
n = 10
fm = nerv.FloatCuMatrix(m, n)
dm = nerv.DoubleCuMatrix(m, n)
for i = 0, m - 1 do
for j = 0, n - 1 do
-- local t = math.random(10)
t = i / (j + 1)
fm[i][j] = t
dm[i][j] = t
end
end
print(fm)
fs = fm:softmax()
print(fs)
print(dm)
ds = dm:softmax()
print(ds)
print(fs + fs)
print(fs - fs)
|