pub fn cat_neighbors_nodes( h_nodes: &Tensor, h_neighbors: &Tensor, e_idx: &Tensor, ) -> Result<Tensor>