def METHOD_NAME(self) -> None:
    net = BasicModel_MultiLayer()
    inp = torch.tensor([[0.0, 5.0, 4.0]])
    self._gradient_input_test_assert(
        net, net.relu, inp, (slice(0, 2, 1),), [[1.0, 1.0, 1.0]]
    )