code: def embed_target_space(target_space_id, hidden_size):
      target_space_emb = common_layers.embedding(
          target_space_id,
          32,
          hidden_size,
          name="target_space_embedding",
      )
      return tf.reshape(target_space_emb, [1, 1, 1, -1])
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from six.moves import range
  from six.moves import zip
  from tensor2tensor.layers import common_attention
  from tensor2tensor.layers import common_hparams
  from tensor2tensor.layers import common_layers
  from tensor2tensor.layers import modalities
  from tensor2tensor.utils import registry
  from tensor2tensor.utils import t2t_model
  import tensorflow.compat.v1
label_desc: Use image node_id 6 for calling a global function with example usage: embed_target_space(target_space_id, hidden_size) and returns: tf