d7efcdf78ef3d0f15b0ac737f049ffc3c2f4fc87,test/augmentation/test_augmentation.py,TestRandomRotation,test_random_rotation,#TestRandomRotation#Any#Any#,1190

Before Change

                              [0., 0., 1., 2.]])  // 4 x 4
        input = input.to(device)

        expected = torch.tensor([[[0.9824, 0.0088, 0.0000, 1.9649],
                                  [0.0000, 0.0029, 0.0000, 0.0176],
                                  [0.0029, 1.0000, 1.9883, 0.0000],
                                  [0.0000, 0.0088, 1.0117, 1.9649]]])  // 1 x 4 x 4
        expected = expected.to(device)

        expected_transform = torch.tensor([[[1.0000, -0.0059, 0.0088],
                                            [0.0059, 1.0000, -0.0088],
                                            [0.0000, 0.0000, 1.0000]]])  // 1 x 3 x 3

After Change

                                            [-0.6731, 0.7396, 1.4002],
                                            [0.0000, 0.0000, 1.0000]]], device=device, dtype=dtype)  // 1 x 3 x 3

        expected_2 = torch.tensor([[[[0.2631, 0.0000, 0.6914, 0.5261],
                                     [0.3457, 0.0000, 0.3235, 0.0000],
                                     [0.0000, 0.7043, 1.6793, 1.0616],
                                     [0.0000, 0.1307, 0.4526, 0.8613]]]], device=device, dtype=dtype)  // 1 x 4 x 4

        out, mat = f(input)
        assert_allclose(out, expected, rtol=1e-6, atol=1e-4)

Instances