docs: def all_near_zero(a: Union[float, complex, Iterable[float], np.ndarray],
                    *,
                    atol: float = 1e-8) -> bool:
      """Checks if the tensor's elements are all near zero.
  
      Args:
          a: Tensor of elements that could all be near zero.
          atol: Absolute tolerance.
      """
      return np.all(np.less_equal(np.abs(a), atol))