code: def set_input_shape(self, input_shape):
      batch_size, dim = input_shape
      self.input_shape = [batch_size, dim]
      self.dim = dim
      self.output_shape = [batch_size, self.num_hid]
      self.make_vars()
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  import tensorflow
  from cleverhans.serial import NoRefModel
label_desc: Use image node_id 2 for calling the Linear obj's underlying member method code with example usage: obj.set_input_shape(input_shape) without return types