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

* break line in several operations

Browse Source
This commit is contained in:
tboquet
2017-02-26 15:09:18 -05:00
parent 0eb0a0b08f
commit b86deb141f
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/backend/tensorflow_backend.py
+2 -2
View File
@@ -97,8 +97,8 @@ def extract_image_patches(x, ksizes, ssizes, border_mode="same",
padding)
# Reshaping to fit Theano
bs, w, h, ch = KTF.int_shape(patches)
patches = tf.reshape(tf.transpose(
tf.reshape(patches, [bs, w, h, -1, ch_i]), [0, 1, 2, 4, 3]),
patches = tf.reshape(patches, [bs, w, h, -1, ch_i])
patches = tf.reshape(tf.transpose(patches, [0, 1, 2, 4, 3]),
[bs, w, h, ch_i, ksizes[0], ksizes[1]])
if dim_ordering == "tf":
patches = KTF.permute_dimensions(patches, [0, 1, 2, 4, 5, 3])