pub fn gather_edges(edges: &Tensor, neighbor_idx: &Tensor) -> Result<Tensor>Expand description
Gather_edges Features [B,N,N,C] at Neighbor indices [B,N,K] => Neighbor features [B,N,K,C]
pub fn gather_edges(edges: &Tensor, neighbor_idx: &Tensor) -> Result<Tensor>Gather_edges Features [B,N,N,C] at Neighbor indices [B,N,K] => Neighbor features [B,N,K,C]