wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-08-18 12:10:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

pep8 fix

Browse Source
This commit is contained in:
Fariz Rahman
2017-04-07 02:47:00 +05:30
committed by GitHub
parent 8491f9271e
commit d6d72aaad3
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/keras_contrib/backend/backend_test.py
+2 -2
View File
@@ -108,7 +108,7 @@ class TestBackend(object):
for rows in [1, 2, 3]:
for cols in [1, 2, 3]:
if K.image_data_format() == 'channels_first':
arr = np.arange(batch_size*channels*scale*scale*rows*cols)\
arr = np.arange(batch_size * channels * scale * scale * rows * cols)\
.reshape((batch_size, channels * scale * scale, rows, cols))
elif K.image_data_format() == 'channels_last':
arr = np.arange(batch_size * rows * cols * scale * scale * channels) \
@@ -120,7 +120,7 @@ class TestBackend(object):
if K.image_data_format() == 'channels_first':
expected = arr.reshape((batch_size, scale, scale, channels, rows, cols))\
.transpose((0, 3, 4, 1, 5, 2))\
.reshape((batch_size, channels, rows*scale, cols*scale))
.reshape((batch_size, channels, rows * scale, cols * scale))
elif K.image_data_format() == 'channels_last':
expected = arr.reshape((batch_size, rows, cols, scale, scale, channels)) \
.transpose((0, 1, 3, 2, 4, 5))\