torch.fft.fftn函数需要哪些参数
时间: 2024-04-27 22:16:20 浏览: 352
Pytorch中torch.nn的损失函数
torch.fft.fftn函数是PyTorch中用于执行多维傅里叶变换的函数。它需要以下参数:
1. input:要进行傅里叶变换的输入张量。它可以是实数或复数类型的张量。
2. signal_ndim:信号的维度。通常为输入张量的维度。
3. normalized:一个布尔值,指定是否对结果进行归一化。默认为False。
4. onesided:一个布尔值,指定是否只返回单边频谱。默认为True。
5. dim:一个整数或元组,指定要在哪些维度上执行傅里叶变换。如果未指定,则在所有维度上执行变换。
阅读全文