Theano 中 a [a < 0] = 0 等价于什么？

Question

Theano 中 NumPy 的 a[a < 0] = 0 等价于什么（张量变量）？我想要所有小于等于零的数字的矩阵元素。

Answer 1

这项工作：

import theano
a=theano.tensor.matrix()
idxs=(a<0).nonzero()
new_a=theano.tensor.set_subtensor(a[idxs], 0)

别忘了，Theano是一种符号语言。所以变量 a 在用户图中没有改变。新变量 new_a 包含新值并且仍然具有旧值。

如果可能，Theano 将优化它以就地工作。

Answer 2

这个也可以，还可以加上上限

import theano
import theano.tensor as T
a = T.matrix()
b = a.clip(0.0)

或者如果你也想要上限，你可能想尝试：

b = T.clip(a, 0.0, 1.0)

其中 1.0 是想要设置上限的位置。

检查文件here

What Is equivalent to a [a < 0] = 0 in Theano?