xnn.common.models.dispersion.pair_edge_keys# xnn.common.models.dispersion.pair_edge_keys(edge_index, shifts, n_atoms)[source]# One int64 key per directed edge from (dst, src, shift), unique per image. Parameters: edge_index (Tensor) shifts (Tensor) n_atoms (int) Return type: Tensor
