如何在 theano 中将小值限制为零

Question

我知道 theano clip 函数，但我想做的是不同的：我有一个向量，如果一个条目很小，但不为零，我想让它为零。大于给定阈值的条目将保持不变。有没有办法在 theano 中做到这一点？

Answer 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]))

如何在 theano 中将小值限制为零

How to clamp small values to zero in theano

theano