[theano-users] Theano flatten doesn't give expected output with arguments ndim

[Lalit Pradhan]

I have variables 'a' and 'b' which are
'theano.sandbox.cuda.var.CudaNdarraySharedVariable'.
I am passing an array of shape ((1,128,300,300)) into 'a'
I am passing an array of shape ((1,1,300,300)) into 'b'
c = a*b . type(c) = 'theano.tensor.var.TensorVariable' of shape
((1,128,300,300))
c = c.flatten(ndim=3), I expect c to be of shape ((1,128,90000)) but it
returns ((1,1,11520000)) instead. Flattening 'a' and 'b' gives expected
results but not 'c'.
Can someone explain the reason for it?