Add new assert_equal test util based on torch.testing.assert_close (#3872)
Add new assert_equal test util based on torch.testing.assert_close (#3872)

Co-authored-by: Philip Meier <8182b159cd16a0fd2cef87181d7c5a58cd09c42c@posteo.de>
pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision,pytorch/vision