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