我正在尝试做fft并在tensorflow模型的层中获取大小，该怎么办？
我尝试使用fft矩阵，但存在一些问题，结果无法满足我
tensor_w256 = tf.convert_to_tensor(w256, dtype=tf.complex64)
x = tf.cast(x, tf.complex64)
x = tf.multiply(tensor_w256, x)
x_mag = tf.cast(x, tf.float32)
所以现在，我想知道如何构造一个张量流层，该层可以完成fft并获得幅值。