enhancementhelp wanted
Métriques du dépôt
- Stars
- (3 433 stars)
- Métriques de merge PR
- (Métriques PR en attente)
Description
Hi,
I can understand the input for pickneglogsoftmax needs to be a vector. If the input is a matrix, I currently have to use select_cols in advance to pick the col vector. Is it the only way to do? Or pickneglogsoftmax may support matrix input in the future?
Thanks!