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 
num_src, num_out = 3, 5
        def make_arg(batch_sizes, n, dim, contig):
            size_arg = batch_sizes[:dim] + (n,) + batch_sizes[dim:]
            return make_tensor(size_arg, d=d, d=d, low=None, high=None, noncontiguous=not contig)
        def ref_index_select(src, dim, idx):
            
            if d == torch.bfloat16:
                src = src.float()
            out = torch.from_numpy(np.take(src.cpu().numpy(), idx.cpu().numpy(), axis=dim))
            if d == torch.bfloat16:
                out = out.to(d=d, d=d)
            return out
        for src_contig, idx_contig in product([True, False], repeat=2):
            for other_sizes in ((), (4, 5)):