mirror of
https://github.com/wassname/keras-contrib.git
synced 2026-10-05 12:50:07 +08:00
* break line in several operations
This commit is contained in:
1 parent
0eb0a0b08f
commit
b86deb141f
1 file changed
+2
-2
@@ -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])
|
||||
|
||||
Reference in new issue
Block a user