def log_prob_from_logits(logits):
"""Softmax function."""
return logits - tf.reduce_logsumexp(logits, keep_dims=True)
微信
公众号
扫码关注公众号
微信
公众号