dmlc--dgl
Hongzhi (Steve), Chen
23d09057e3
* [Misc] Black auto fix. * sort Co-authored-by: Steve <ubuntu@ip-172-31-34-29.ap-northeast-1.compute.internal>
9 行
237 B
Python
9 行
237 B
Python
import torch.nn as nn
|
|
|
|
|
|
def GlorotOrthogonal(tensor, scale=2.0):
|
|
if tensor is not None:
|
|
nn.init.orthogonal_(tensor.data)
|
|
scale /= (tensor.size(-2) + tensor.size(-1)) * tensor.var()
|
|
tensor.data *= scale.sqrt()
|