If you do weight decay, you can do this: model.target_layer.weight.data.var(1) to get the variance of the output units’ weight.
If you don’t do weight decay, and have to check the unit activation, look up Module.register_forward_hook 132. This allows you to check the output of intermediate layers quite conveniently.