code: def test_scatter_tensor():
      ab = torch.zeros(2, 1)
  
      a, b = scatter(ab, chunks=2)
  
      assert a.tensor.size() == (1, 1)
      assert b.tensor.size() == (1, 1)
  
file_path: fairscale/tests/nn/pipe/test_microbatch.py
imports: import pytest
  import torch
  import torch.cuda
  from fairscale.nn.pipe.microbatch import Batch, check, gather, scatter
label_desc: Use image node_id 9 for calling a global function with example usage: test_scatter_tensor() without return types