0
点赞
收藏
分享

微信扫一扫

pytorch的scatter函数(二维)

贵州谢高低 2022-03-31 阅读 51

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

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

dim=0:

 dim=1:

举报

相关推荐

0 条评论