aboutsummaryrefslogblamecommitdiff
path: root/layer/affine.lua
blob: 5f1b4ce4a2158dcd7b15ec7770e4d7631c41a16a (plain) (tree)
1
2
3
4
5
6
7
8
9
10
11










                                                                                 













                                                       
local LinearTransParam = nerv.class('nerv.LinearTransParam', 'nerv.Param')
local BiasParam = nerv.class('nerv.BiasParam', 'nerv.LinearTransParam')
local AffineLayer = nerv.class('nerv.AffineLayer', 'nerv.Layer')

function LinearTransParam:read(pcdata)
    self.trans = nerv.CuMatrixFloat.new_from_host(nerv.MMatrixFloat.load(pcdata))
end

function LinearTransParam:write(pfhandle)
    self.trans:new_to_host():save(pfhandle)
end

function AffineLayer:__init(id, ltp, bp)
    self.ltp = ltp
    self.bp = bp
end

function nerv.AffineLayer:update(input, output)
end

function nerv.AffineLayer:propagate(input, output)
end

function nerv.AffineLayer:back_propagate(input, output)
end