torch.unbind
torch.unbind(input, dim=0) -> seq
Removes a tensor dimension.
Returns a tuple of all slices along a given dimension, already
   without it.
Parameters:
      * input (Tensor) -- the tensor to unbind
  * **dim** (*int*) -- dimension to remove

Example:
  >>> torch.unbind(torch.tensor([[1, 2, 3],
  >>>                            [4, 5, 6],
  >>>                            [7, 8, 9]]))
  (tensor([1, 2, 3]), tensor([4, 5, 6]), tensor([7, 8, 9]))
