根据布尔条件将 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]
我正在尝试将 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]