wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-09-09 11:25:17 +08:00
Code Issues Packages Projects Releases Wiki Activity

pep8 fix

Browse Source
This commit is contained in:
Fariz Rahman
2017-04-07 02:48:37 +05:30
committed by GitHub
parent d6d72aaad3
commit 671ca018a9
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/backend/theano_backend.py
+2 -2
View File
@@ -190,10 +190,10 @@ def depth_to_space(input, scale, data_format=None):
input = _preprocess_conv2d_input(input, data_format)
b, k, row, col = input.shape
out_channels = k // (scale**2)
out_channels = k // (scale ** 2)
x = T.reshape(input, (b, scale, scale, out_channels, row, col))
x = T.transpose(x, (0, 3, 4, 1, 5, 2))
out = T.reshape(x, (b, out_channels, row*scale, col*scale))
out = T.reshape(x, (b, out_channels, row * scale, col * scale))
out = _postprocess_conv2d_output(out, input, None, None, None, data_format)
return out