所以一句话概括gather 操作就是:根据 index ,在 input 的 dim 维度上收集value。 具体来说, input 就是源tensor,等会我们要在这个tensor 上执行gather ...
確定! 回上一頁