radial_basis

radial_basis#

Radial basis functions for distance encoding.

prxteinmpnn.utils.radial_basis

prxteinmpnn.utils.radial_basis.compute_radial_basis(backbone_coordinates, neighbor_indices)[source]#

Compute the radial basis functions for backbone coordinates.

Return type:

Float[Array, 'L K R D']

Parameters:

  • backbone_coordinates (Float[Array, '4 3 3'])

  • neighbor_indices (Int[Array, 'num_atoms num_neighbors'])

Contents