2016-01-03 40 views
2

在我的程序中,我想使用張量的形狀值。例如,張量x具有(3,4,5)的形狀。我想要得到的4的價值,我想通過以下方式:在theano中獲取張量的值

t=x.shape[1] #returns a scalar 
t=x.shape[1].eval() #returns a array(4) 

我怎樣才能得到的值4,我想要的是一個int類型,不標或其他類型的。

回答

0

試試這個:

import numpy as np 
int(np.asscalar(x.shape[1].eval())) 
+0

我使用x.shape [1] .eval()[1],並得到4最後 – hunter

相關問題