Is there some straightforward way to reshape while using add_module() ?
ATM I have to resort to
class RESHAP(nn.Module):
	def __init__(self, nz):
		super(RESHAP, self).__init__()
		self.nz = nz

	def forward(self, input):
		return input.view(-1, self.nz, 1, 1)

	def __repr__(self):
		return self.__class__.__name__ + ' ()'

main.add_module(‘initial.{0}.reshape’.format(nz/4), RESHAP(nz/4))