enhancementhelp wanted
仓库指标
- Star
- (3,433 star)
- PR 合并指标
- (PR 指标待抓取)
描述
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!