Function torch_assert_equal

tests/unit/util.py:129–134

Compare two tensors or non-tensor numbers for their equality. Add msg=blah to add an additional comment to when assert fails.

(actual, expected, **kwargs)

def torch_assert_equal(actual, expected, **kwargs) -> None:
    """
    Compare two tensors or non-tensor numbers for their equality.
    Add msg=blah to add an additional comment to when assert fails.
    """
    torch.testing.assert_close(actual, expected, rtol=0.0, atol=0.0, **kwargs)


def torch_assert_close(actual, expected, **kwargs) -> None: