Fix logits_to_logprobs for 2-D and 3-D logits (#1002)
* Fix logits_to_logprobs for 2-D and 3-D logits * Set dtype to single * Test size
K
kddubey committed
5a8944672f24974fc6e80fecc2c22555e97dbb04
Parent: 534b1ea
Committed by GitHub <noreply@github.com>
on 12/16/2023, 11:59:26 PM