- if cond(kernel_size, stride):
- n = (input_size - kernel_size) // stride
- if n * stride + kernel_size == input_size:
- q = conv_chain(n + 1, output_size, depth - 1, cond)
+ if cond(depth, kernel_size, stride):
+ n = (input_size - kernel_size) // stride + 1
+ if (n - 1) * stride + kernel_size == input_size:
+ q = conv_chain(n, output_size, depth - 1, cond)