Which initialization would you want to perform?
From Andrew’s slides it looks like you sample from a normal distribution (torch.randn) and multiply with different scales, i.e.
torch.sqrt(2/n_in), torch.tanh(torch.sqrt(1/n_in)), and torch.sqrt(2/(n_in+n_out)).
You can find the initialization methods like xavier_normal here 70.