如何在theano

时间:2016-01-05 13:23:16

标签: theano

我知道theano剪辑功能,但我想做的是不同的:我有一个向量,如果一个条目很小,但不是零,我想把它变为零。大于给定阈值的条目将保持不变。 有没有办法在theano中做到这一点?

1 个答案:

答案 0 :(得分:1)

您可以使用以下代码段将特定阈值以下的矢量值剪辑为零:

import theano
import theano.tensor as T

x = T.ivector('x')
threshold = 5  # change accordingly
x_clipped = x * (x > threshold)

f = theano.function(inputs=[x], outputs=x_clipped)

print(f([1, 2, 3, 4, 5, 6, 7, 8, 9, 10]))