xnn.common.models.dispersion_fast.atm.plan_triplets

xnn.common.models.dispersion_fast.atm.plan_triplets(edge_index, n_atoms)

Sort the edges by (center, neighbor) and lay out the kernel launches.

Parameters:

• edge_index (Tensor)

• n_atoms (int)

Return type:

TripletPlan