code: def test_mask_select():
      src = torch.randn(6, 8)
      mask = torch.tensor([False, True, False, True, False, True])
  
      out = mask_select(src, 0, mask)
      assert out.size() == (3, 8)
      assert torch.equal(src[torch.tensor([1, 3, 5])], out)
  
      jit = torch.jit.script(mask_select)
      assert torch.equal(jit(src, 0, mask), out)
  
imports: import torch
  from torch_geometric.utils import index_to_mask, mask_select, mask_to_index
label_desc: Use image node_id 1 for calling a global function with example usage: test_mask_select() without return types