Skip to content

Support tuple of axis in softmax_cross_entropy_with_integer_labels #5148

Support tuple of axis in softmax_cross_entropy_with_integer_labels

Support tuple of axis in softmax_cross_entropy_with_integer_labels #5148

Annotations

1 warning

Python 3.9 on ubuntu-latest jax=0.4.27

succeeded Jan 8, 2025 in 19m 56s