I have tensor which looks like
tensor([[1., 1., 0., 1., 0., 0., 1., 1., 0., 1.],
        [1., 1., 0., 1., 0., 0., 1., 1., 0., 0.],
        [1., 1., 0., 1., 0., 0., 0., 0., 0., 1.]])

Whats the best way to find the indices of last one in each row? In the above case, it should return [9, 7, 9]