Google I/O returns May 18-20! Reserve space and build your schedule Register now

tfg.geometry.transformation.axis_angle.is_normalized

Determines if the axis-angle is normalized or not.

tfg.geometry.transformation.axis_angle.is_normalized(
    axis, angle, atol=0.001, name='axis_angle_is_normalized'
)

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.
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.