3 # Any copyright is dedicated to the Public Domain.
4 # https://creativecommons.org/publicdomain/zero/1.0/
6 # Written by Francois Fleuret <francois@fleuret.org>
10 from torch import Tensor
12 ######################################################################
14 def minimal_input_size(w, layer_specs):
15 assert w > 0, 'The input is too small'
19 kernel_size, stride = layer_specs[0]
20 v = int(math.ceil((w - kernel_size) / stride)) + 1
21 v = minimal_input_size(v, layer_specs[1:])
22 return (v - 1) * stride + kernel_size
24 ######################################################################
28 if __name__ == "__main__":
30 layer_specs = [ (17, 5), (5, 4), (3, 2), (3, 2) ]
34 for kernel_size, stride in layer_specs:
35 layers.append(nn.Conv2d(1, 1, kernel_size, stride))
37 for kernel_size, stride in reversed(layer_specs):
38 layers.append(nn.ConvTranspose2d(1, 1, kernel_size, stride))
40 m = nn.Sequential(*layers)
42 h = minimal_input_size(240, layer_specs)
43 w = minimal_input_size(320, layer_specs)
45 x = Tensor(1, 1, h, w).normal_()
47 print(x.size(), m(x).size())
49 ######################################################################