在theano中逐行标准化矩阵

时间:2015-02-10 08:50:36

标签: python matrix normalization theano

假设我有一个大小为N的矩阵n_i x n_o,我想逐行归一化,即, 每行的总和应为1。我怎么能在theano做到这一点?

动机:使用softmax为我返回错误,所以我尝试通过实现我自己的softmax版本来回避它。

2 个答案:

答案 0 :(得分:9)

查看以下内容是否对您有用:

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)

答案 1 :(得分:4)

或者您也可以使用

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