def __call__(self, x):
    if self.dr:
        x = F.dropout(x, self.dr)
    x = self.l(x)
    x = self.activation(x)
    if self.out_shape:
        x = F.reshape(x, self.out_shape)
    return x
def __call__(self, x):
    if self.dr:
        x = F.dropout(x, self.dr)
    x = self.l(x)
    x = self.activation(x)
    if self.out_shape:
        x = F.reshape(x, self.out_shape)
    return x