(self)
| 50 | self.assertEqual(test_tensor_tuple[i].requires_grad, initial_grads[i]) |
| 51 | |
| 52 | def test_gradient_basic(self) -> None: |
| 53 | model = BasicModel() |
| 54 | input = torch.tensor([[5.0]], requires_grad=True) |
| 55 | input.grad = torch.tensor([[9.0]]) |
| 56 | grads = compute_gradients(model, input)[0] |
| 57 | assertTensorAlmostEqual(self, grads, [[0.0]], delta=0.01, mode="max") |
| 58 | # Verify grad attribute is not altered |
| 59 | assertTensorAlmostEqual(self, input.grad, [[9.0]], delta=0.0, mode="max") |
| 60 | |
| 61 | def test_gradient_basic_2(self) -> None: |
| 62 | model = BasicModel() |
nothing calls this directly
no test coverage detected