scatter() 和 scatter_() 的作用一样，区别在于 scatter() 不会直接修改原来的 Tensor， scatter_() 会直接在原tensor的基础上修改。
PyTorch 中，函数加上下划线代表直接在原来的 Tensor 上修改。
dim=0:
dim=1：
pytorch的scatter函数(二维)
阅读 51
2022-03-31
scatter() 和 scatter_() 的作用一样，区别在于 scatter() 不会直接修改原来的 Tensor， scatter_() 会直接在原tensor的基础上修改。
PyTorch 中，函数加上下划线代表直接在原来的 Tensor 上修改。
dim=0:
dim=1：
相关推荐
精彩评论（0）
