给定百万个 0~999 的 ID 和对应权重,如何用 CUDA 按 ID 累加为 1000 个结果?
Sorry o(╥﹏╥)o,暂无参考答案。带这道题,问问AI吧~
问 AI
本题目录
相关面经
1 篇公司未注明
AI InfraEmbedding Sparse Feature Pooling:A是 100 万个离散 ID(0~999),B是 100 万个float,计算个长度为1000的float数组C,C[i] = \sum_{j s.t. A[j] = i} B[j]
Sorry o(╥﹏╥)o,暂无参考答案。带这道题,问问AI吧~
Embedding Sparse Feature Pooling:A是 100 万个离散 ID(0~999),B是 100 万个float,计算个长度为1000的float数组C,C[i] = \sum_{j s.t. A[j] = i} B[j]