pytorch的scatter函数(二维)

阅读 52

2022-03-31

        scatter()scatter_() 的作用一样,区别在于 scatter() 不会直接修改原来的 Tensor, scatter_() 会直接在原tensor的基础上修改。

        PyTorch 中,函数加上下划线代表直接在原来的 Tensor 上修改。

dim=0:

 dim=1:

相关推荐

精彩评论(0)

0 0 举报