`nn.int.ones_(module.weight)` is always more readable than `nn.init.constant_(module.weight, 1)`, and for `nn.init.zeros_` as well.