ff_layers.py 文件源码

def fprop(self, all_states):
        shape0 = all_states.shape[0]
        single_frame = all_states.max(0)

        if self.ntimes:
            single_frame = TT.shape_padleft(all_states.max(0))
            mask = TT.alloc(numpy.float32(1),
                        shape0, *[1 for k in xrange(all_states.ndim-1)])
            rval = single_frame * mask
            self.out = rval
            return rval
        self.out = single_frame
        return single_frame