fix dtype mismatch in `nn.cross_entropy_loss`

Jittor / jittor

Jittor is a high-performance deep learning framework based on JIT compiling and meta-operators.

https://cg.cs.tsinghua.edu.cn/jittor/

Apache License 2.0

3.07k stars 307 forks source link

Closed CHEN-Xinsheng closed 1 month ago