提交 b7a6a2c6 authored 作者: Razvan Pascanu's avatar Razvan Pascanu 提交者: Pascal Lamblin

use floatX

上级 b07912fc
...@@ -270,14 +270,18 @@ class test_RopLop(RopLop_checker): ...@@ -270,14 +270,18 @@ class test_RopLop(RopLop_checker):
filter_shape = (2, 2, 2, 3) filter_shape = (2, 2, 2, 3)
image_dim = len(image_shape) image_dim = len(image_shape)
filter_dim = len(filter_shape) filter_dim = len(filter_shape)
input = tensor.TensorType('float64', [False] * input = tensor.TensorType(
image_dim)(name='input') theano.config.floatX,
filters = tensor.TensorType('float64', [False] * [False] * image_dim)(name='input')
filter_dim)(name='filter') filters = tensor.TensorType(
ev_input = tensor.TensorType('float64', [False] * theano.config.floatX,
image_dim)(name='ev_input') [False] * filter_dim)(name='filter')
ev_filters = tensor.TensorType('float64', [False] * ev_input = tensor.TensorType(
filter_dim)(name='ev_filters') theano.config.floatX,
[False] * image_dim)(name='ev_input')
ev_filters = tensor.TensorType(
theano.config.floatX,
[False] * filter_dim)(name='ev_filters')
bsize = image_shape[0] bsize = image_shape[0]
if image_dim != 3: if image_dim != 3:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论