如何查找torch.sum的axis参数文档？PyTorch隐藏参数查询

1. 函数签名里的*和axis的关系 #

那个*是Python语法，用来区分位置参数和关键字专属参数，和axis本身无关。但torch.sum的dim参数（也就是你用的axis）属于*之后的关键字参数，网页版文档可能只展示了核心参数，没把别名列出来。

2. axis参数的实际身份 #

axis是dim参数的别名，PyTorch为了兼容NumPy等其他框架，给所有维度相关的函数都提供了这个别名——比如torch.mean、torch.max都支持用axis代替dim，两者功能完全一致。网页版文档通常优先显示dim，但不代表axis不合法。

3. 查找这类「隐藏」参数的方法 #

• 直接在Python环境查看完整文档：执行print(torch.sum.__doc__)或者help(torch.sum)，会输出包含所有参数别名的完整说明，比网页版更全面。
• 参考PyTorch官方的参数别名规范：PyTorch有统一的参数别名规则，比如dim/axis、keepdim/keepdims都是通用的别名对，这类规则在官方核心文档里有说明。
• 查看源码定义：你找到的native_functions.yaml里，会明确标注参数的别名，比如torch.sum的定义中会说明dim的别名是axis。

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