5 from torch import Tensor
7 ######################################################################
9 def minimal_input_size(w, layer_specs):
10 assert w > 0, 'The input is too small'
15 w = math.ceil((w - k) / s) + 1
16 w = minimal_input_size(w, layer_specs[1:])
17 return int((w - 1) * s + k)
19 ######################################################################
21 layer_specs = [ (11, 5), (5, 2), (3, 2), (3, 2) ]
25 layers.append(nn.Conv2d(1, 1, l[0], l[1]))
27 for l in reversed(layer_specs):
28 layers.append(nn.ConvTranspose2d(1, 1, l[0], l[1]))
30 m = nn.Sequential(*layers)
32 h = minimal_input_size(240, layer_specs)
33 w = minimal_input_size(320, layer_specs)
35 x = Tensor(1, 1, h, w).normal_()
37 print(x.size(), m(x).size())