


torch.vsplit(t, 2)
      (tensor([[0., 1., 2., 3.],
               [4., 5., 6., 7.]]),
       tensor([[ 8.,  9., 10., 11.],
               [12., 13., 14., 15.]]))
      >>> torch.vsplit(t, [3, 6])
      (tensor([[ 0.,  1.,  2.,  3.],
               [ 4.,  5.,  6.,  7.],
               [ 8.,  9., 10., 11.]]),
       tensor([[12., 13., 14., 15.]]),
       tensor([], size=(0, 4)))


