PyTorch中scatter()函数如何应用在笔记中？

本文共计727个文字，预计阅读时间需要3分钟。

scatter() 和 scatter_() 的作用类似，但 scatter() 不会直接修改原始的 Tensor，而 scatter_() 会。scatter_() 在 PyTorch 中通常用于在原始 Tensor 上直接进行修改，相当于 scatter(dim, index, src) 函数的参数应用。