示例#1
0
 def message_and_aggregate(self, adj_t: SparseTensor, x):
     assert adj_t is not None
     return spmm(adj_t, x.view(-1, 1)).view(-1)
示例#2
0
 def message_and_aggregate(self, adj_t: SparseTensor,
                           x: OptPairTensor) -> Tensor:
     return spmm(adj_t, x[0], reduce=self.aggr)
示例#3
0
 def message_and_aggregate(self, adj_t: SparseTensor, x):
     return spmm(adj_t, x.view(-1, 1)).view(-1)