scatter() 和 scatter_() 的作用一样,区别在于 scatter() 不会直接修改原来的 Tensor, scatter_() 会直接在原tensor的基础上修改。
PyTorch 中,函数加上下划线代表直接在原来的 Tensor 上修改。
dim=0:
dim=1:
scatter() 和 scatter_() 的作用一样,区别在于 scatter() 不会直接修改原来的 Tensor, scatter_() 会直接在原tensor的基础上修改。
PyTorch 中,函数加上下划线代表直接在原来的 Tensor 上修改。
dim=0:
dim=1: