Update DFT definition to more closely align with PyTorch by enabling axis attribute, and arbitrary tensor rank. (#10842)
* Add axis attribute
* fix breaks
* Enable axis-specified DFT
* remove static cast
Co-authored-by: Sheil Kumar <sheilk@microsoft.com>