def METHOD_NAME(self):
    def f(x):
        a = x.sum(dim=1)
        b = x.sum(dim=1, keepdim=False)
        c = x.sum(dim=1, keepdim=False)
        d = x.sum(dim=1)
        return a + b + c + d
    t = torch.randn(2, 2)
    check(self, f, t, 3)