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)