python - tensorflow/keras 神经网络中的复杂输入/输出是可能的吗？

Question

我听说可以为使用 tensorflow/keras 构建的神经网络提供复杂数据，并获得指定每个层的“dtype”= tf.complex64(或类似)的复杂输出。

X = K.Input(shape = (n_taps,1), dtype = tf.complex64)
fc1 = K.layers.LSTM(n_fc1,activation="tanh",dtype = tf.complex64)  (X)

每个单层的声明都没有给出错误，但是调用第二层作为参数给出第一层(我的意思是第二行)给出了以下预期错误:

TypeError: Input 'b' of 'MatMul' Op has type float32 that does not match type complex64 of argument 'a'.

我不明白是否可能有这种网络。有没有人有更多关于这个的信息？提前致谢

Answer 1

这是因为 K.layers.LSTM 不支持 tf.complex64 类型。您需要自己实现该层。您应该从子类化 tf.keras.layers.Layer 开始，您可以在 tf.keras.layers.Layer 的文档中找到一个示例.该示例称为 SimpleDense 层。