脚本宝典收集整理的这篇文章主要介绍了torch 与索引相关函数,脚本宝典觉得挺不错的,现在分享给大家,也给大家做个参考。
根据索引查找值
torch.gather() https://pytorch.org/docs/stable/generated/torch.gather.htML#torch.gather
根据索引修改函数的值
torch.scatter() https://pytorch.org/docs/stable/generated/torch.Tensor.scatter_.html#torch.Tensor.scatter_
以上是脚本宝典为你收集整理的torch 与索引相关函数全部内容,希望文章能够帮你解决torch 与索引相关函数所遇到的问题。
本图文内容来源于网友网络收集整理提供,作为学习参考使用,版权属于原作者。
如您有任何意见或建议可联系处理。小编QQ:384754419,请注明来意。