def METHOD_NAME(a: Tensor, b: Tensor, rtol: float = 1e-3, atol: float = 1e-3):
    assert_close(a,
                 b,
                 rtol=rtol,
                 atol=atol,
                 msg=f"Tensor not close, shape: {a.shape} vs {b.shape}, \
                                                   dtype: {a.dtype} vs {b.dtype}")