def max_pool_concat(self, hs):
num_output = len(hs[0])
houts = []
i = 0
x = F.max(F.dstack([h[i] for h in hs]),2)
houts.append(x)
for i in range(1,num_output):
#x = 0
#for h in hs:
# x = x + h[i]
x = F.concat([h[i] for h in hs],1)
houts.append(x) # Merged branch exit and main exit
return houts
评论列表
文章目录