1. 概述
gather 和 scatter 是 PyTorch 里非常容易混淆、但其实可以用一个非常直观的方式理解的两个操作。
先记住一句话：
gather = 按照 index"取东西"
scatter = 按照 index"放东西"
2. gather
待续。。。。
gather和scatter API
Tancenter2026-09-21 11:30
相关推荐
程序猿阿森5 小时前
混合精度详解：FP16 vs BF16，Loss Scaling 与 PyTorch 实战Είναι η κοπέλα1 天前
模型量化完全指南：GGUF、GPTQ、AWQ 怎么选Soonyang Zhang2 天前
根据时戳从pytorch trace中获取kernel信息承渊政道2 天前
【从零开始大模型开发与微调:基于PyTorch与ChatGLM】(开源大模型ChatGLM使用详解)129Lab3 天前
电池热管理仿真的AI加速：用Python+PINN物理信息神经网络替代传统CFD的可行性探索Είναι η κοπέλα3 天前
llama.cpp 与 GGUF 格式：本地大模型的“裸引擎“头发够用的程序员3 天前
TensorRT 自定义算子插件实战（三）：手搓 2×2 最大池化 customMaxpoolfor_ever_love__4 天前
PyTorch 张量与 autograd——自动求导怎么工作Είναι η κοπέλα5 天前
显存计算与模型选择：你的显卡能跑多大的模型海天一色y6 天前
模型部署的「通用语」：ONNX 转换