Pytorch lstm initialization

Print it