theano 地图和减少

theano.map 和 theano.scan_module.reduce 是 theano_scan 的包装。他们可以被视为 scan 的残疾人版本。你可以查看基本扫描用法部分以供参考。

import theano
import theano.tensor as T
s_x = T.ivector()
s_sqr, _ = theano.map(
    fn = lambda x:x*x,
    sequences = [s_x])
s_sum, _ = theano.reduce(
    fn = lambda: x,y:x+y,
    sequences = [s_x],
    outputs_info = [0])
fn = theano.function([s_x], [s_sqr, s_sum])
fn([1,2,3,4,5]) #[1,4,9,16,25], 15