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

tfg.geometry.transformation.axis_angle.is_normalized

Determines if the axis-angle is normalized or not.

tfg.geometry.transformation.axis_angle.is_normalized(
    axis: type_alias.TensorLike,
    angle: type_alias.TensorLike,
    atol: float = 0.001,
    name: str = 'axis_angle_is_normalized'
) -> 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.
atol The absolute tolerance parameter.
name A name for this op that defaults to "axis_angle_is_normalized".

A tensor of shape [A1, ..., An, 1], where False indicates that the axis is not normalized.