Esempio n. 1
0
def test_log_softmax_manual(t: Tensor) -> None:
    np.testing.assert_allclose(ep.log_softmax(t).exp().numpy(),
                               ep.softmax(t).numpy(),
                               rtol=1e-6)
Esempio n. 2
0
def test_log_softmax(t: Tensor) -> Tensor:
    return ep.log_softmax(t)