求问pytroch操作

level 1

c13lib 楼主

原数据 B * N * K * M * H 索引是 idx B * N * 1(K中选的一个)
需要做成 B * N * 1 * M * H
感觉scatter gather select都不适合

2024年08月02日 04点08分 1

level 11

sealed_ss

可以用mean或max等操作 [乖]

2024年08月03日 05点08分 2