viable/strict/1786915540: Add handling for empty batch dimensions in FFT operations (#190483)
PyTorch 在 2026-08-16 发布的版本中修复了 FFT 操作对空批次维度张量的处理。此前，在 MPS、CUDA 和 x86 CPU 后端，对空张量执行 FFT 会失败：MPSGraph 拒绝空占位符，cuFFT 报 CUFFT_INVALID_SIZE，oneMKL DFTI 报配置错误，且元注册在追踪时无法计算空输出。
Development
- First Reportviable/strict/1786915540: Add handling for empty batch dimensions in FFT operations (#190483)PyTorch Core
- Current AssessmentPyTorch 持续修复边缘情况，反映了深度学习框架在追求稳健性和全面性方面的努力。此类修复对依赖 PyTorch 进行大规模训练或推理的团队至关重要，尤其是在处理动态形状或数据并行时可能遇到空批次。Agent Pulse · analysis
PyTorch 在 2026-08-16 的版本中修复了 FFT 操作对空批次维度张量的处理。此前，在 MPS、CUDA 和 x86 CPU 后端，对空张量执行 FFT 会失败：MPSGraph 拒绝空占位符，cuFFT 报 CUFFT_INVALID_SIZE，oneMKL DFTI 报配置错误，且元注册在追踪时无法计算空输出。该修复解决了这些后端无法规划或执行零元素张量变换的问题，并处理了追踪下形状推断对空输出的处理。
该修复表明 PyTorch 正在完善对空张量（零元素）的算子支持，这涉及后端规划、形状推断和元注册的协同。对于 FFT 这类需要底层库支持的算子，空批次维度需要特殊处理，以避免后端拒绝或错误。这提示开发者在使用动态形状或空批次时需注意算子兼容性。
PyTorch 持续修复边缘情况，反映了深度学习框架在追求稳健性和全面性方面的努力。此类修复对依赖 PyTorch 进行大规模训练或推理的团队至关重要，尤其是在处理动态形状或数据并行时可能遇到空批次。
该修复提升了 PyTorch 的稳定性和可靠性，减少了因空批次导致的崩溃或错误，对依赖 PyTorch 的 AI 应用开发和部署具有积极影响，降低了维护成本。
未来，PyTorch 可能会继续扩展对空张量的支持，覆盖更多算子，并改进元注册和追踪逻辑。开发者可关注后续版本中关于空张量处理的更新。