def __call__(self, x):
    if self.dr:
        x = F.dropout(x, self.dr)
    x = F.transpose(x, (0, 2, 1, 3))
    out_shape = list(x.shape)
    x = F.reshape(x, ((- 1), (x.shape[2] * x.shape[3])))
    x = self.l(x)
    x = self.activation(x)
    out_shape[2] = self.out_ch
    x = F.reshape(x, out_shape)
    x = F.transpose(x, (0, 2, 1, 3))
    return x
def __call__(self, x):
    if self.dr:
        x = F.dropout(x, self.dr)
    x = F.transpose(x, (0, 2, 1, 3))
    out_shape = list(x.shape)
    x = F.reshape(x, ((- 1), (x.shape[2] * x.shape[3])))
    x = self.l(x)
    x = self.activation(x)
    out_shape[2] = self.out_ch
    x = F.reshape(x, out_shape)
    x = F.transpose(x, (0, 2, 1, 3))
    return x