Save the date! Google I/O returns May 18-20 Register now

tfg.geometry.transformation.quaternion.from_axis_angle

Converts an axis-angle representation to a quaternion.

tfg.geometry.transformation.quaternion.from_axis_angle(
    axis, angle, name='quaternion_from_axis_angle'
)

Note:

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.