Beispiel #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)
Beispiel #2
0
 def message_and_aggregate(self, adj_t: SparseTensor,
                           x: OptPairTensor) -> Tensor:
     return spmm(adj_t, x[0], reduce=self.aggr)
Beispiel #3
0
 def message_and_aggregate(self, adj_t: SparseTensor, x):
     return spmm(adj_t, x.view(-1, 1)).view(-1)