


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


