def __call__(self, x, train=True):
links = self.children()
h = F.leaky_relu(next(links)(x))
for link in links:
h = link(h, train)
return h
微信
公众号
扫码关注公众号