ValueError異常：NEW_ORDER [2]是1，但輸入只有1軸

Question

對於下面的代碼段：

imgs = theano.shared(numpy.asarray(numpy.random.uniform((batch_size,c,n1,n2)),dtype=theano.config.floatX)) 
imgs = imgs.dimshuffle((0,'x', 1, 2, 3)) 

我得到以下錯誤：

ValueError: new_order[2] is 1, but the input only has 1 axes. 

我沒有看到問題恰恰在哪裏。任何想法？

Answer 1

你忘了告訴是什麼參數(batch_size,c,n1,n2)

imgs = theano.shared(numpy.asarray(numpy.random.uniform(size=(batch_size,c,n1,n2)),dtype=theano.config.floatX)) 
imgs = imgs.dimshuffle((0,'x', 1, 2, 3))