根据boolen条件将theano张量转换为0和1的张量

时间:2017-03-19 02:53:44

标签: python theano

我正在尝试将theano张量转换为0和1的张量,其值基于>这是正确的方法吗?

pip install -U pyautogui

由于

1 个答案:

答案 0 :(得分:0)

您正在寻找命令gt(代表大于)。它的工作原理如下:

from theano import tensor as T
X = T.dvector('X')
f = theano.function([X],T.gt(X,0))
x = np.random.randint(0,10,5)
print x  # prints [4 0 5 9 5]
print f(x)  # prints [True False True True True]