给定百万个 0~999 的 ID 和对应权重,如何用 CUDA 按 ID 累加为 1000 个结果?

未标记

Sorry o(╥﹏╥)o,暂无参考答案。带这道题,问问AI吧~

问 AI
本题目录

相关面经

1 篇

公司未注明

AI Infra
查看原面经 →
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]

相关题目

4 题