For example, If I have a 2D tensor of [[1,3],[2,1]], how can I get indices that value 1 has occurre ? (it should give [0,0] and [1,1])
[[1,3],[2,1]]
1
I too