Rename keep_dims to keepdims in reduce_logsumexp
@@ -108,7 +108,7 @@ def get_state_shape_invariants(tensor):
def _log_prob_from_logits(logits):
- return logits - tf.reduce_logsumexp(logits, axis=2, keep_dims=True)
+ return logits - tf.reduce_logsumexp(logits, axis=2, keepdims=True)
def compute_batch_indices(batch_size, beam_size):
