fix: use jnp.exp(log_probs) instead of softmax(log_probs) in compute_entropy_from_logits#1387
Open
kuishou68 wants to merge 1 commit intogoogle:mainfrom
Open
fix: use jnp.exp(log_probs) instead of softmax(log_probs) in compute_entropy_from_logits#1387kuishou68 wants to merge 1 commit intogoogle:mainfrom
kuishou68 wants to merge 1 commit intogoogle:mainfrom
