How to initialize weight for LSTM?

[Zhao_Wulanaren]

My initialization is showed as following:

But I want to initialize the weights with Xavier not randn. Does someone know how to do it?

[Kaixhin (Kai Arulkumaran)]

Use torch.nn.init.xavier_uniform or torch.nn.init.xavier_normal, whichever one you’re looking for.

[Zhao_Wulanaren]

Thank you very much!