YouTipOutput result is:
Non-zero element indices: tensor([[0, 1], [1, 0], [2, 1]])
Pytorch Torch Nonzero
# Return the indices of non-zero elements
indices = torch.nonzero(x)
print("Non-zero element indices:")
print(indices)