Thanks for tuning in to Google I/O. View all sessions on demandWatch on demand

tfg.geometry.transformation.axis_angle.from_euler

Converts Euler angles to an axis-angle representation.

tfg.geometry.transformation.axis_angle.from_euler(
    angles: type_alias.TensorLike,
    name: str = 'axis_angle_from_euler'
) -> Tuple[tf.Tensor, tf.Tensor]

The conversion is performed by first converting to a quaternion representation, and then by converting the quaternion to an axis-angle.

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

angles A tensor of shape [A1, ..., An, 3], where the last dimension represents the three Euler angles. [A1, ..., An, 0] is the angle about x in radians [A1, ..., An, 1] is the angle about y in radians and [A1, ..., An, 2] is the angle about z in radians.
name A name for this op that defaults to "axis_angle_from_euler".

A tuple of two tensors, respectively of shape [A1, ..., An, 3] and [A1, ..., An, 1], where the first tensor represents the axis, and the second represents the angle. The resulting axis is a normalized vector.