calculate convolution dimensions

calc_deconv_dims.py

x = 80 # spectrogram scale
output = 24000  # waveform scale
max_f = 50
max_s = 50
N = 2

# calculate function
calc_conv_dim = lambda x, f, s, p: (x - f + p) // s + 1
calc_deconv_dim = lambda x, f, s, p: int((x - 1) * s - p + f)

combination_f = lambda: [(f, s) for f in range(0, max_f, 2) for s in range(0, max_s, 2)]

# get all of the combination
combination = combination_f()
for _ in range(N - 1):
    combination = [[*x, *y] for x in combination for y in combination_f()]
	
# match all
for item in combination:
    x_ = x
    for n in range(0, N):
        f, s = item[n * 2], item[n * 2 + 1]
        x_ = calc_deconv_dim(x_, f, s, 0)
    if x_ == output:
        print(item)