From 551a0531c941d190b482e8a9decf67afdf6168d1 Mon Sep 17 00:00:00 2001 From: "siyeong.lee" Date: Mon, 25 Mar 2019 18:30:35 +0900 Subject: [PATCH] fix typo error in layers.py docs formula --- se2cnn/layers.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/se2cnn/layers.py b/se2cnn/layers.py index 38067ea..f33e108 100644 --- a/se2cnn/layers.py +++ b/se2cnn/layers.py @@ -327,7 +327,7 @@ def rotate_gconv_kernels(kernel, periodicity=2 * np.pi, diskMask=True): kernels_temp = kernels_planar_rotated[orientation] # [kernelSizeH,kernelSizeW,channelsIN,channelsOUT,orientations_nb] kernels_temp = tf.transpose(kernels_temp, [0, 1, 3, 4, 2]) - # [kernelSizeH*kernelSizeW*channelsIN*channelsOUT*orientations_nb] + # [kernelSizeH*kernelSizeW*channelsIN*channelsOUT,orientations_nb] kernels_temp = tf.reshape( kernels_temp, [kernelSizeH * kernelSizeW * channelsIN * channelsOUT, orientations_nb]) # Roll along the orientation axis