获取theano中tensor的值
Get the value of tensor in theano
在我的程序中,我想使用张量的形状值。例如,张量 x 的形状为 (3,4,5)。我想得到4的值,我尝试了下面的方法:
t=x.shape[1] #returns a scalar
t=x.shape[1].eval() #returns a array(4)
如何得到4的值,我要的是int类型,不是标量或其他类型。
试试这个:
import numpy as np
int(np.asscalar(x.shape[1].eval()))
在我的程序中,我想使用张量的形状值。例如,张量 x 的形状为 (3,4,5)。我想得到4的值,我尝试了下面的方法:
t=x.shape[1] #returns a scalar
t=x.shape[1].eval() #returns a array(4)
如何得到4的值,我要的是int类型,不是标量或其他类型。
试试这个:
import numpy as np
int(np.asscalar(x.shape[1].eval()))