def conv_and_pool(self, x, conv):
x = F.relu(conv(x)).squeeze(3) # (N,Co,W)
x = F.max_pool1d(x, x.size(2)).squeeze(2)
return x
微信
公众号
扫码关注公众号