VolumetricFullConvolution_bug.lua

require 'cunn'

--test cuda & non-cuda version
function testVolumetricFullConvolution()
  local input = torch.rand(1,2,10,10,10) * 2 - 1
  local a = nn.VolumetricFullConvolution(2,3, 3,3,3, 1,1,1)
  local b = nn.VolumetricFullConvolution(2,3, 3,3,3, 1,1,1)
  b:cuda()
  
  b.weight = a.weight:cuda() 
  b.bias = a.bias:cuda()
  
  local out1 = a:forward(input)
  local out2 = b:forward(input:cuda()):double()
  
  print("VolumetricFullConvolution: " .. (out1-out2):abs():sum())
end

function testSpatialFullConvolution()
  local input = torch.rand(1,2,10,10) * 2 - 1
  local a = nn.SpatialFullConvolution(2,3, 3,3, 1,1)
  local b = nn.SpatialFullConvolution(2,3, 3,3, 1,1)
  b:cuda()
  
  b.weight = a.weight:cuda() 
  b.bias = a.bias:cuda()
  
  local out1 = a:forward(input)
  local out2 = b:forward(input:cuda()):double()
  
  print("SpatialFullConvolution: " .. (out1-out2):abs():sum())
end

testSpatialFullConvolution()
testVolumetricFullConvolution()

--[[ prints on my machine something like:
SpatialFullConvolution: 5.9598770636923e-06
VolumetricFullConvolution: 1150.674565644	]]