code: def softmax_last_timestep_class_label_top(body_output,
                                            targets,
                                            model_hparams,
                                            vocab_size):
    
    del targets  # unused arg
    with tf.variable_scope(
        "softmax_last_timestep_onehot_class_label_modality_%d_%d" % (
            vocab_size, model_hparams.hidden_size)):
      x = body_output
      x = tf.expand_dims(x[:, -1], 1)  # Pick the last timestep
      return tf.layers.dense(x, vocab_size)
docs: Loss for class label.