Platforms: linux

This test was disabled because it is failing on master ([recent examples](http://torch-ci.com/failure/test_torch.py%3A%3ATestTorch%3A%3Atest_equal)).

The cause seems to point to https://github.com/pytorch/pytorch/pull/100024 (or D45282119 as it was landed internally)

cc @mruberry @houseroad @ezyang 
def make_prob_dist(shape, is_contiguous):
            if is_contiguous:
                if d == torch.half:
                    return torch.zeros(shape, d=d).uniform_().to(d=torch.half)
                return torch.zeros(shape, d=d, d=d).uniform_()
            elif len(shape) == 1:
                if d == torch.half:
                    return torch.zeros((shape + [5]), d=d).uniform_().to(d=torch.half)[:, 2]
                return torch.zeros((shape + [5]), d=d, d=d).uniform_()[:, 2]
            else:
                
                new_shape = [2, shape[1], 7, 1, shape[0], 1, 10]
                if d == torch.half:
                    prob_dist = torch.zeros(new_shape, d=d).uniform_().to(d=torch.half)
                else:
                    prob_dist = torch.zeros(new_shape, d=d, d=d).uniform_()
                prob_dist = prob_dist.transpose(1, 4)
                prob_dist = prob_dist[1, :, 5, 0, :, 0, 4]
                assert not prob_dist.is_contiguous()  
                return prob_dist
        for is_contiguous in (True, False):
            
            n_row = 3
            for n_col in range(4, 5 + 1):
                prob_dist = make_prob_dist([n_row, n_col], is_contiguous)
                
                zero_prob_indices = torch.LongTensor(n_row).random_(-2, n_col).tolist()
                for i, j in enumerate(zero_prob_indices):