Created by: zzs95

GATNE/model.py

trans_weights = fl.create_parameter(
        shape=[
            self.edge_type_count, self.embedding_u_size,
            self.embedding_size // self.att_head
        ],
        attr=fluid.initializer.TruncatedNormalInitializer(
            loc=0.0, scale=1.0 / math.sqrt(self.embedding_size)),
        dtype='float32',
        name='trans_w')

attr应该是default_initializer 吧