multiinputsequential.py 文件源码

def max_pool(self, hs):
        num_output = len(hs[0]) 
        houts = []
        for i in range(num_output):
            shape = hs[0][i].shape
            h = F.dstack([F.reshape(h[i],(shape[0], -1)) for h in hs])
            x = 1.0*F.max(h,2)
            x = F.reshape(x, shape)
            houts.append(x)
        return houts