python - Theano 元素明智的最大值

Question

我正在努力寻找值(value)s=max(ele, 0) 在 theano 的矩阵上逐元素。我对 theano 没有太多经验。

目前为止

x = theano.tensor.dmatrix('x')
s = (x + abs(x)) / 2  # poor man's trick
linmax = function([x], s)

这行得通，但不是很漂亮，我想我应该可以为此使用 theano.tensor.maximum。

在 matlab 中，为了做我想做的，我会写linmax=@(x) max (x, zeros (size (x)))

Answer 1

这对我有用:

import theano.tensor as T
from theano import function

x = T.dmatrix('x')
linmax = function([x], T.maximum(x,0))

测试:

linmax([[-1,-2],[3,4]])

输出:

array([[0.,0.],[3.,4.]])