python - 如何使用 torch.fft.fft2 输出与 troch.fft 相同的结果？

Question

在pytorch 1.1.0的文档中，对返回的描述torch.fft是

“将实部和虚部一起作为相同形状输入的张量返回”

在 pytorch1.8.1 中，torch.fft 被替换为torch.fft.fft2，并torch.fft.fft2以复数形式输出结果

对于相同的数据，输出torch.fft为

张量([32779.2891])张量([67.1836])张量([11.9802])张量([-47.5729])张量([-101.7718])张量([-5.7989])张量([17.5259])张量([41.0830] ) 张量([-5.1960]) 张量([42.2860])

的输出结果torch.fft.fft2是

张量(32779.2891+0.j)张量(36.2154-25.2860j)张量(22.9704-39.8544j)张量(-62.9149+6.8637j)张量(-87.2122-94.2708j)张量(-3.2740+52.0396666-32. +46.9949j)张量(-50.1910-30.1725j)张量(-8.8877+19.0709j)张量(39.9689-32.3084j)

输入是由 生成的 256x256 数据 input = numpy.random.rand(256,256) ，上述结果是使用此数据（256x256）作为输入生成的数据的一部分。使用此数据（256x256），Matlab 和 torch.fft.fft2 输出相同的结果

torch.fft 如何将结果的实部和虚部存储在相同形状的张量中？

如何使用torch.fft.fft2输出相同的结果torch.fft？

标签： pythonpytorch