def METHOD_NAME(self, adj_t: SparseTensor,
                          x: PairTensor) -> Tensor:
    if isinstance(adj_t, SparseTensor):
        adj_t = adj_t.set_value(None, layout=None)
    return spmm(adj_t, x[0], reduce=self.aggr)