tfep.utils.geometry.get_axis_from_name

tfep.utils.geometry.get_axis_from_name(name: Literal['x', 'y', 'z']) → Tensor

Return the 3D vector representation of an axis.

Parameters:

name (Literal[‘x’, ‘y’, ‘z’]) – The name of the axis.

Returns:

axis – Shape (3,). The unit vector representation of the axis.

Return type:

torch.Tensor