enhancementhelp wanted
Métricas do repositório
- Stars
- (3.433 stars)
- Métricas de merge de PR
- (Métricas PR pendentes)
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!