什么相当于Theano中的[a <0] = 0？

Question

什么相当于a[a < 0] = 0Theano中的NumPy (张量变量)？我想要所有小于数字的矩阵元素等于零.

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将优化此功能以便在现场工作.