352ebffaa5bd74a454c3b2b10121b2d959ade2fc,opennmt/models/sequence_tagger.py,SequenceTagger,_build,#SequenceTagger#Any#Any#Any#Any#,49

Before Change

        // predictions must contain tensors with the same batch size
        // so replicate the transition matrix accordingly.
        transition_params = tf.convert_to_tensor(transition_params)
        transition_params = tf.expand_dims(transition_params, axis=0)
        transition_params = tf.tile(transition_params, [tf.shape(logits)[0], 1, 1])

        predictions["logits"] = logits
        predictions["transition_params"] = transition_params
      else:
        probs = tf.nn.softmax(logits)
        predictions["argmax"] = tf.argmax(probs, axis=2)

After Change

      predictions = {}
      predictions["length"] = encoder_sequence_length
      predictions["labels"] = labels_vocab_rev.lookup(labels)

      return tf.estimator.EstimatorSpec(
        mode,

Instances