使用tensorflow tf.nn.conv2d和float64输入和输出

时间:2017-06-26 17:57:55

标签: python tensorflow

我正在尝试使用tensorflow运行物理模拟。我发现tf.nn.conv2d只接受float32输入。我对此的理解是,对于使用float64的机器学习应用程序不会导致任何类型的改进,因为GPU通常使用float32。有没有办法破解tf.nn.conv2d使其使用float64,即使我只需要在CPU上运行模型?或者有替代tf.nn.conv2d,我可以使用float64

0 个答案:

没有答案