在theano中按行劃分矩陣規範化

Question

假設我有一個大小爲n_i x n_o的矩陣N，我想按行逐級規範化，即 每行的總和應該是1。我該如何在theano中做到這一點？

動機：使用softmax返回給我的錯誤，所以我嘗試通過實現我自己的softmax版本來避開它。

Answer 1

看看以下是對您有用：

import theano 
import theano.tensor as T 

m = T.matrix(dtype=theano.config.floatX) 
m_normalized = m/m.sum(axis=1).reshape((m.shape[0], 1)) 

f = theano.function([m], m_normalized) 

import numpy as np 
a = np.exp(np.random.randn(5, 10)).astype(theano.config.floatX) 

b = f(a) 
c = a/a.sum(axis=1)[:, np.newaxis] 

from numpy.testing import assert_array_equal 
assert_array_equal(b, c) 

Answer 2

，或者您也可以使用

m/m.norm(1, axis=1).reshape((m.shape[0], 1))