如何在tensorflow 2.0.0中导入fft2d？

import tensorflow as tf
from tensorflow.python.ops.signal import fft_ops

# 创建一个 2D 张量
input_tensor = tf.constant([[1.0, 2.0], [3.0, 4.0]], dtype=tf.complex64)

# 使用 fft2d 函数计算二维傅里叶变换
output_tensor = fft_ops.fft2d(input_tensor)

print(output_tensor)

input_tensor = tf.constant([[1.0, 2.0], [3.0, 4.0]], dtype=tf.float32)
input_tensor_complex = tf.complex(input_tensor, tf.zeros_like(input_tensor))
output_tensor = fft_ops.fft2d(input_tensor_complex)

import tensorflow as tf
from tensorflow.signal import fft2d

input_tensor = tf.constant([[1.0, 2.0], [3.0, 4.0]], dtype=tf.complex64)
output_tensor = fft2d(input_tensor)
print(output_tensor)