如何在PyTorch源码中查找函数定义与实现？以torch.real为例

• 从核心原语模块入手：你找到的torch/_prims/__init__.py是正确方向，这里是PyTorch核心原语（primitives）的注册入口，第803行的real相关定义是该操作的元信息与接口声明，是上层API的底层支撑。
• 追踪API绑定逻辑：PyTorch的多数API不会直接在torch/__init__.py中定义，而是通过torch._ops或torch.library的注册机制，将底层实现绑定到torch命名空间。可以查看torch/_refs/__init__.py，这里有real的Python参考实现，再通过注册映射到更底层的逻辑。
• 定位C内核实现：若要查看最底层的计算逻辑，可前往aten/src/ATen/native目录，找到复数相关文件（如ComplexOps.cpp），里面包含real算子的C实现——复数张量的实部提取属于基础操作，必然有对应的内核代码。
• 利用内置工具快速定位：在Python环境中执行以下操作获取线索：
  • 打印torch.real.__module__，查看该函数所属模块；
  • 执行torch._C._jit_get_operation('aten::real')，直接关联到对应的ATen算子，进而追踪到C++代码；
  • 查看torch.real.__doc__中的注释，往往会包含实现相关的提示。
• 理清API分层逻辑：PyTorch API分为多层：上层torch命名空间API、中间primitives/refs层、ATen C++算子层、后端（CUDA/CPU）实现层。torch.real作为上层API，通过注册机制关联底层实现，因此不会直接出现在torch.__init__.py中。

内容的提问来源于stack exchange，提问作者kingwales