import torchimport torch.nn as nnimport torch.nn.functional as Fclass Block(nn.Module):
    '''Depthwise conv + Pointwise conv'''
    def