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