tfg.geometry.transformation.quaternion.from_axis_angle

Converts an axis-angle representation to a quaternion.

tfg.geometry.transformation.quaternion.from_axis_angle(
    axis: type_alias.TensorLike,
    angle: type_alias.TensorLike,
    name: str = 'quaternion_from_axis_angle'
) -> tf.Tensor

In the following, A1 to An are optional batch dimensions.

axis A tensor of shape [A1, ..., An, 3], where the last dimension represents a normalized axis.
angle A tensor of shape [A1, ..., An, 1], where the last dimension represents an angle.
name A name for this op that defaults to "quaternion_from_axis_angle".

A tensor of shape [A1, ..., An, 4], where the last dimension represents a normalized quaternion.

ValueError If the shape of axis or angle is not supported.