build_csr_neighbor_list¶ torch_sim.transforms.build_csr_neighbor_list(mapping, system_mapping, shifts_idx, n_atoms)[source]¶ Build CSR inputs for kernels that require neighbors grouped by center atom. Parameters: mapping (Tensor) system_mapping (Tensor) shifts_idx (Tensor) n_atoms (int) Return type: tuple[Tensor, Tensor, Tensor]