pytorch

343c1c21 - update nn.init.calculate_gain doc example

update nn.init.calculate_gain doc example Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/20131 Differential Revision: D15218126 Pulled By: ezyang fbshipit-source-id: 164c9d1573cd4d2c3689fb83b952e71862d4f1f2