removing the requirement for providing size in nn.Mul