deeperwin.utils.get_distance_matrix

deeperwin.utils.get_distance_matrix(r_el)[source]

Compute distance matrix omitting the main diagonal (i.e. distance to the particle itself)

Parameters

r_el – [batch_dims x n_electrons x 3]

Returns

differences [batch_dims x n_el x n_el x 3], distances [batch_dims x n_el x n_el]

Return type

tuple