Hi,
You can do the following that makes use of automatic expansion for element wise operations when they have the same number of dimensions:
beta=nn.Parameter(torch.zeros(32))
gamma=nn.Parameter(torch.ones(32))

y = x * gamma.view(1,32,1,1) + beta.view(1,32,1,1)