fixed batch TemporalConvolution bug

1 files changed, 4 insertions, 4 deletions

diff --git a/test/test.lua b/test/test.lua
index c945d4e..ff4e649 100644
--- a/test/test.lua
+++ b/test/test.lua
@@ -1388,13 +1388,13 @@ function nntest.TemporalConvolution()
    local outputGrad = torch.randn(output:size())
    local inputGrad = module:backward(input, outputGrad):clone()
    
-   local input1D = input:select(1, 1)
+   local input1D = input:select(1, 2)
    local output1D = module:forward(input1D)
-   local outputGrad1D = outputGrad:select(1, 1)
+   local outputGrad1D = outputGrad:select(1, 2)
    local inputGrad1D = module:backward(input1D, outputGrad1D)
    
-   mytester:assertTensorEq(output:select(1,1), output1D, 0.000001, 'error on 2D vs 1D forward)')
-   mytester:assertTensorEq(inputGrad:select(1,1), inputGrad1D, 0.000001, 'error on 2D vs 1D backward)')
+   mytester:assertTensorEq(output:select(1,2), output1D, 0.000001, 'error on 2D vs 1D forward)')
+   mytester:assertTensorEq(inputGrad:select(1,2), inputGrad1D, 0.000001, 'error on 2D vs 1D backward)')
 end
 
 function nntest.TemporalSubSampling()