From d8ca85f445c7fc20d5d64cab7da402b78f817f0d Mon Sep 17 00:00:00 2001 From: Michael Oliver Date: Wed, 22 Feb 2017 18:23:39 -0800 Subject: [PATCH] fix tf dim ordering test --- tests/keras_contrib/layers/test_convolutional.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/keras_contrib/layers/test_convolutional.py b/tests/keras_contrib/layers/test_convolutional.py index 483d5cb..1d4bd68 100644 --- a/tests/keras_contrib/layers/test_convolutional.py +++ b/tests/keras_contrib/layers/test_convolutional.py @@ -121,7 +121,7 @@ def test_cosineconvolution_2d(): elif dim_ordering == 'tf': X = np.random.randn(1, 5, 5, 3) input_dim = (5, 5, 3) - W0 = X[:, ::-1, ::-1, :] + W0 = X[0, ::-1, ::-1, :, None] model = Sequential() model.add(convolutional.CosineConvolution2D(1, 5, 5, bias=True, input_shape=input_dim))