def __call__(self, x):
h = F.relu(self.bn1(self.conv1(x)))
h = F.max_pooling_2d(h, 3, stride=2, pad=0)
return h
微信
公众号
扫码关注公众号
微信
公众号