Fangction

@wushilian thank you. And I have another question. In your code W = tf.Variable(tf.zeros([128, 20])) b = tf.Variable(initial_value=[-1, -0.2, -0.5, -0.35, 0, -0.5, 0.5, -0.67, 1, -0.8, -1, 0.8, -0.5,...

@wushilian thank you again. And I also want to ask one more question. Do you think the STN's output image size must be fixed size？I see you define the output...