bilinear_bug.lua

gistfile1.txt

require('nn')
require('nngraph')

local m

do
    local x1, x2 = nn.Identity()(), nn.Identity()()
    local y = nn.Bilinear(2, 2, 2){x1, x2}
    m = nn.gModule({x1, x2}, {y})
end

local x1 = torch.Tensor(1, 2):fill(1)
local x2 = x1:clone()
local z  = x1:clone()

print(m:forward{x1, x2})
print(m:backward({x1, x2}, z))

m:clearState()

print(m:forward{x1, x2})
print(m:backward({x1, x2}, z))