pytorch/torch/testing
Philip Meier dbf3451c6e Add support for checking tensor containers in torch.testing (#55385)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/55385

This renames `assert_tensors_(equal|close)` to `_check_tensors_(equal|close)` and exposes two new functions: `assert_(equal|close)`. In addition to tensor pairs, the newly added functions also support the comparison of tensors in sequences or mappings. Otherwise their signature stays the same.

Test Plan: Imported from OSS

Reviewed By: albanD

Differential Revision: D27903805

Pulled By: mruberry

fbshipit-source-id: 719d19a1d26de8d14cb25846e3d22a6ac828c80a
2021-04-24 23:36:36 -07:00
..
_internal Add tests for CUDAFuture (#56518) 2021-04-24 07:07:31 -07:00
__init__.py Un-ignore F403 in .flake8 (#55838) 2021-04-13 09:24:07 -07:00
_asserts.py Add support for checking tensor containers in torch.testing (#55385) 2021-04-24 23:36:36 -07:00
_check_kernel_launches.py make torch.testing asserts importable (#54769) 2021-04-07 23:53:02 -07:00
_core.py OpInfo porting for torch.real & torch.imag (#55134) 2021-04-15 13:28:21 -07:00