transform.CountNeighbors

class transform.CountNeighbors(*args: Any, **kwargs: Any)

Store the number of neighbors for each atom

Parameters:

sorted – Set to false if chosen neighbor list yields unsorted center indices (idx_i).