复数的张量流运算

时间:2018-12-23 19:56:01

标签: python tensorflow machine-learning

我有一个包含复数的数据集。当我将数据输入网络时,出现了错误:

ValueError: An initializer for variable encoder/conv2d/kernel of <dtype: 'complex64'> is required

以下是我网络中的一些代码:

self.input_placeholder=tf.placeholder(tf.complex64,[None,self.train_data[0].shape[1],self.train_data[0].shape[2],self.train_data[0].shape[3]])

在初始化所有参数之前,卷积步骤中会发生错误:

layer=tf.layers.conv2d(inputs,64,[1,self.F],strides=(1,1),padding='same',activation=None)

有什么解决办法吗?

张量流中是否支持复数?

非常感谢您!

1 个答案:

答案 0 :(得分:1)

尚不支持复杂的初始化程序。

这里有一个开放的问题描述功能请求:

https://github.com/tensorflow/tensorflow/issues/17097

根据该票证中的讨论,看来Keras已经提供了一种方法。也许您可以做类似的事情。