code: def build(self, input_shape):
      self.lstm_cell = tf.nn.rnn_cell.LSTMCell(
          self.out_dim, initializer=self.initializer
      )
      self.initial_state = lstm_cell.zero_state(
          input_shape[0], dtype=tf.float32
      )
      self.built = True
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import collections
  import tensorflow
  from tensorflow.python.util import nest
label_desc: Use image node_id 2 for calling the LSTMLayer obj's underlying member method code with example usage: obj.build(input_shape) without return types