Keras使用numpy包含theano.config.floatX浮点类型的数组.这可以在您的.theanorc文件中配置.
通常,它将float64用于CPU计算和float32GPU计算,但float32如果您愿意,也可以将其设置为在CPU上工作时.
您可以通过命令创建正确类型的零填充数组
X = numpy.zeros((4,3), dtype=theano.config.floatX)
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
2454 次 |
| 最近记录: |