如何强制Theano按列执行softmax？

Question

我有一个2D数组，我想应用softmax函数列。尝试以下方法：

value = numpy.array([[1.0,2.0], [3.0,9.0], [7.0,1.0]], dtype=theano.config.floatX)
m = theano.shared(value, name='m', borrow=True)
y = theano.tensor.nnet.softmax(m)
print y.eval()

因此，我得到以下输出：

^{pr2}$

这意味着操作已按行应用。有没有办法强迫Theano按列顺序执行softmax？在

Answer 1

简单的解决方案：转置

>>> import theano as th
>>> T = th.tensor
>>> x = th.tensor.matrix()
>>> y = T.nnet.softmax(x.T).T
>>> fn = th.function([x],y)
>>> fn([[1.,2.],[1.,3.]])
array([[ 0.5       ,  0.26894143],
       [ 0.5       ,  0.7310586 ]], dtype=float32)

对于张量rank>；=3，转置需要指定轴。在

^{pr2}$