def __init__(self, adj_channel_num, initializer='zeros', **kwargs): self.adj_channel_num = adj_channel_num self.initializer = initializer if enabled_bspmm: import kgcn.bspmm_call as bspmm self.bspmm_obj = bspmm.BatchedSpMM() if enabled_bconv: import kgcn.bconv_call as bconv self.bconv_obj = bconv.BatchedConv() if enabled_batched: import kgcn.batched_call as batched self.bspmdt_obj = batched.BatchedSpMDT() super(GINAggregate, self).__init__(**kwargs)
def __init__(self, output_dim, adj_channel_num, initializer='glorot_uniform', **kwargs): self.output_dim = output_dim self.adj_channel_num = adj_channel_num self.initializer = initializer if enabled_bspmm: import kgcn.bspmm_call as bspmm self.bspmm_obj = bspmm.BatchedSpMM() if enabled_bconv: import kgcn.bconv_call as bconv self.bconv_obj = bconv.BatchedConv() if enabled_batched: import kgcn.batched_call as batched self.bspmdt_obj = batched.BatchedSpMDT() super(GraphConv, self).__init__(**kwargs)