Remove constraint that last dimension is forced to be 1 in cross_entropy (#19606)
* Remove constraint that last dimension is forced to be 1 in cross_entropy test=develop * modify labels last dims test=develop
Showing
想要评论请 注册 或 登录
* Remove constraint that last dimension is forced to be 1 in cross_entropy test=develop * modify labels last dims test=develop