class_docstr: Simple embedding model.
class_docstr_tok: ["Simple","embedding","model","."]
code: def call(self, inputs, training=None):
      # The shape of the sequence output from the embedding layer is
      # [batch_size, sequence_length, embedding_size]
      sequence_output = self.embed_layer(inputs)
      first_token_tensor = tf.squeeze(
          sequence_output[:, 0:1, :], axis=1
      )
      # The shape of the pooled output from the embedding layer is
      # [batch_size, embedding_size]
      pooled_output = self.pool_layer(first_token_tensor)
      return sequence_output, pooled_output
  
imports: from absl.testing import parameterized
  import numpy
  import tensorflow
  from tensorflow import estimator
  from tensorflow_privacy.privacy.optimizers import dp_optimizer_keras
  from tensorflow_privacy.privacy.optimizers import dp_optimizer_keras_vectorized
label_desc: Use image node_id 2 for calling the SimpleEmbeddingModel obj's underlying member method code with example usage: obj.call(inputs, training) and returns: sequence_output,  pooled_output