pub fn gather_nodes(nodes: &Tensor, neighbor_idx: &Tensor) -> Result<Tensor>Expand description
Gather Nodes
Features [B,N,C] at Neighbor indices [B,N,K] => [B,N,K,C] Flatten and expand indices per batch [B,N,K] => [B,NK] => [B,NK,C]
pub fn gather_nodes(nodes: &Tensor, neighbor_idx: &Tensor) -> Result<Tensor>Gather Nodes
Features [B,N,C] at Neighbor indices [B,N,K] => [B,N,K,C] Flatten and expand indices per batch [B,N,K] => [B,NK] => [B,NK,C]