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

convert a theano tensor into a tensor of 0 and 1 based on a boolen condition

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

tensor = tensor > 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]