# Example 1 ip_tensor=torch.tensor([[1,2,3],[4,5,6]]) torch.unbind(ip_tensor,dim=0) # Output (tensor([1, 2, 3]), tensor([4, 5, 6]))