def compute_output_shape(self, input_shape):
    return (input_shape.as_list()[:(- 1)] + [self.output_dim])
def compute_output_shape(self, input_shape):
    return (input_shape.as_list()[:(- 1)] + [self.output_dim])