def __init__(self, input_shape, output_shape):
    self.out_H = output_shape[0]
    self.out_W = output_shape[1]
    self.out_D = output_shape[2]
    self.input_shape = input_shape
def __init__(self, input_shape, output_shape):
    self.out_H = output_shape[0]
    self.out_W = output_shape[1]
    self.out_D = output_shape[2]
    self.input_shape = input_shape