def __init__(self, target_shape, **kwargs):
    self.target_shape = target_shape
    super(TotalReshape, self).__init__(**kwargs)
def __init__(self, target_shape, **kwargs):
    self.target_shape = target_shape
    super(TotalReshape, self).__init__(**kwargs)