概述
总说
一个非常有用的函数，主要是用于“group index”的操作。
先安装一下 https://github.com/rusty1s/pytorch_scatter
from torch_scatter import scatter
import torch
src = (torch.rand(2, 6, 2)*4).int()
最后
以上就是机智小霸王为你收集整理的Pytorch学习 （二十六）---- torch.scatter的使用总说的全部内容，希望文章能够帮你解决Pytorch学习 （二十六）---- torch.scatter的使用总说所遇到的程序开发问题。
如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。
本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。
发表评论 取消回复