


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



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